mirror of
https://github.com/lfnovo/open-notebook.git
synced 2026-05-05 15:24:35 +00:00
b7e656a319
New front-end Launch Chat API Manage Sources Enable re-embedding of all contents Sources can be added without a notebook now Improved settings Enable model selector on all chats Background processing for better experience Dark mode Improved Notes Improved Docs: - Remove all Streamlit references from documentation - Update deployment guides with React frontend setup - Fix Docker environment variables format (SURREAL_URL, SURREAL_PASSWORD) - Update docker image tag from :latest to :v1-latest - Change navigation references (Settings → Models to just Models) - Update development setup to include frontend npm commands - Add MIGRATION.md guide for users upgrading from Streamlit - Update quick-start guide with correct environment variables - Add port 5055 documentation for API access - Update project structure to reflect frontend/ directory - Remove outdated source-chat documentation files
222 lines
No EOL
7 KiB
Python
222 lines
No EOL
7 KiB
Python
from typing import Any, Dict, List
|
|
|
|
from fastapi import APIRouter, HTTPException
|
|
from loguru import logger
|
|
from pydantic import BaseModel, Field
|
|
|
|
from open_notebook.domain.podcast import SpeakerProfile
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
class SpeakerProfileResponse(BaseModel):
|
|
id: str
|
|
name: str
|
|
description: str
|
|
tts_provider: str
|
|
tts_model: str
|
|
speakers: List[Dict[str, Any]]
|
|
|
|
|
|
@router.get("/speaker-profiles", response_model=List[SpeakerProfileResponse])
|
|
async def list_speaker_profiles():
|
|
"""List all available speaker profiles"""
|
|
try:
|
|
profiles = await SpeakerProfile.get_all(order_by="name asc")
|
|
|
|
return [
|
|
SpeakerProfileResponse(
|
|
id=str(profile.id),
|
|
name=profile.name,
|
|
description=profile.description or "",
|
|
tts_provider=profile.tts_provider,
|
|
tts_model=profile.tts_model,
|
|
speakers=profile.speakers
|
|
)
|
|
for profile in profiles
|
|
]
|
|
|
|
except Exception as e:
|
|
logger.error(f"Failed to fetch speaker profiles: {e}")
|
|
raise HTTPException(
|
|
status_code=500,
|
|
detail=f"Failed to fetch speaker profiles: {str(e)}"
|
|
)
|
|
|
|
|
|
@router.get("/speaker-profiles/{profile_name}", response_model=SpeakerProfileResponse)
|
|
async def get_speaker_profile(profile_name: str):
|
|
"""Get a specific speaker profile by name"""
|
|
try:
|
|
profile = await SpeakerProfile.get_by_name(profile_name)
|
|
|
|
if not profile:
|
|
raise HTTPException(
|
|
status_code=404,
|
|
detail=f"Speaker profile '{profile_name}' not found"
|
|
)
|
|
|
|
return SpeakerProfileResponse(
|
|
id=str(profile.id),
|
|
name=profile.name,
|
|
description=profile.description or "",
|
|
tts_provider=profile.tts_provider,
|
|
tts_model=profile.tts_model,
|
|
speakers=profile.speakers
|
|
)
|
|
|
|
except HTTPException:
|
|
raise
|
|
except Exception as e:
|
|
logger.error(f"Failed to fetch speaker profile '{profile_name}': {e}")
|
|
raise HTTPException(
|
|
status_code=500,
|
|
detail=f"Failed to fetch speaker profile: {str(e)}"
|
|
)
|
|
|
|
|
|
class SpeakerProfileCreate(BaseModel):
|
|
name: str = Field(..., description="Unique profile name")
|
|
description: str = Field("", description="Profile description")
|
|
tts_provider: str = Field(..., description="TTS provider")
|
|
tts_model: str = Field(..., description="TTS model name")
|
|
speakers: List[Dict[str, Any]] = Field(..., description="Array of speaker configurations")
|
|
|
|
|
|
@router.post("/speaker-profiles", response_model=SpeakerProfileResponse)
|
|
async def create_speaker_profile(profile_data: SpeakerProfileCreate):
|
|
"""Create a new speaker profile"""
|
|
try:
|
|
profile = SpeakerProfile(
|
|
name=profile_data.name,
|
|
description=profile_data.description,
|
|
tts_provider=profile_data.tts_provider,
|
|
tts_model=profile_data.tts_model,
|
|
speakers=profile_data.speakers
|
|
)
|
|
|
|
await profile.save()
|
|
|
|
return SpeakerProfileResponse(
|
|
id=str(profile.id),
|
|
name=profile.name,
|
|
description=profile.description or "",
|
|
tts_provider=profile.tts_provider,
|
|
tts_model=profile.tts_model,
|
|
speakers=profile.speakers
|
|
)
|
|
|
|
except Exception as e:
|
|
logger.error(f"Failed to create speaker profile: {e}")
|
|
raise HTTPException(
|
|
status_code=500,
|
|
detail=f"Failed to create speaker profile: {str(e)}"
|
|
)
|
|
|
|
|
|
@router.put("/speaker-profiles/{profile_id}", response_model=SpeakerProfileResponse)
|
|
async def update_speaker_profile(profile_id: str, profile_data: SpeakerProfileCreate):
|
|
"""Update an existing speaker profile"""
|
|
try:
|
|
profile = await SpeakerProfile.get(profile_id)
|
|
|
|
if not profile:
|
|
raise HTTPException(
|
|
status_code=404,
|
|
detail=f"Speaker profile '{profile_id}' not found"
|
|
)
|
|
|
|
# Update fields
|
|
profile.name = profile_data.name
|
|
profile.description = profile_data.description
|
|
profile.tts_provider = profile_data.tts_provider
|
|
profile.tts_model = profile_data.tts_model
|
|
profile.speakers = profile_data.speakers
|
|
|
|
await profile.save()
|
|
|
|
return SpeakerProfileResponse(
|
|
id=str(profile.id),
|
|
name=profile.name,
|
|
description=profile.description or "",
|
|
tts_provider=profile.tts_provider,
|
|
tts_model=profile.tts_model,
|
|
speakers=profile.speakers
|
|
)
|
|
|
|
except HTTPException:
|
|
raise
|
|
except Exception as e:
|
|
logger.error(f"Failed to update speaker profile: {e}")
|
|
raise HTTPException(
|
|
status_code=500,
|
|
detail=f"Failed to update speaker profile: {str(e)}"
|
|
)
|
|
|
|
|
|
@router.delete("/speaker-profiles/{profile_id}")
|
|
async def delete_speaker_profile(profile_id: str):
|
|
"""Delete a speaker profile"""
|
|
try:
|
|
profile = await SpeakerProfile.get(profile_id)
|
|
|
|
if not profile:
|
|
raise HTTPException(
|
|
status_code=404,
|
|
detail=f"Speaker profile '{profile_id}' not found"
|
|
)
|
|
|
|
await profile.delete()
|
|
|
|
return {"message": "Speaker profile deleted successfully"}
|
|
|
|
except HTTPException:
|
|
raise
|
|
except Exception as e:
|
|
logger.error(f"Failed to delete speaker profile: {e}")
|
|
raise HTTPException(
|
|
status_code=500,
|
|
detail=f"Failed to delete speaker profile: {str(e)}"
|
|
)
|
|
|
|
|
|
@router.post("/speaker-profiles/{profile_id}/duplicate", response_model=SpeakerProfileResponse)
|
|
async def duplicate_speaker_profile(profile_id: str):
|
|
"""Duplicate a speaker profile"""
|
|
try:
|
|
original = await SpeakerProfile.get(profile_id)
|
|
|
|
if not original:
|
|
raise HTTPException(
|
|
status_code=404,
|
|
detail=f"Speaker profile '{profile_id}' not found"
|
|
)
|
|
|
|
# Create duplicate with modified name
|
|
duplicate = SpeakerProfile(
|
|
name=f"{original.name} - Copy",
|
|
description=original.description,
|
|
tts_provider=original.tts_provider,
|
|
tts_model=original.tts_model,
|
|
speakers=original.speakers
|
|
)
|
|
|
|
await duplicate.save()
|
|
|
|
return SpeakerProfileResponse(
|
|
id=str(duplicate.id),
|
|
name=duplicate.name,
|
|
description=duplicate.description or "",
|
|
tts_provider=duplicate.tts_provider,
|
|
tts_model=duplicate.tts_model,
|
|
speakers=duplicate.speakers
|
|
)
|
|
|
|
except HTTPException:
|
|
raise
|
|
except Exception as e:
|
|
logger.error(f"Failed to duplicate speaker profile: {e}")
|
|
raise HTTPException(
|
|
status_code=500,
|
|
detail=f"Failed to duplicate speaker profile: {str(e)}"
|
|
) |