Upload 67 files

src/agents/agent_transcript/func.py

@@ -14,6 +14,7 @@ from pydantic import BaseModel, Field
 
 
 class State(TypedDict):
+    language: str
     script_structure_analyzer_prompt: str
     comment_insight_extractor_prompt: str
     scientific_fact_finder_prompt: str
@@ -141,10 +142,17 @@ Comment Insight Extractor Response: {state["comment_insight_extractor_response"]
 
 
 def script_re_outline(state: State):
+
+    language = state["language"]
+    if not language:
+        language = "English"
+
     response = chain_script_re_outline.invoke(
         {
             "messages": state["messages"],
-            "prompt": state["script_re_outline_prompt"],
+            "prompt": state["script_re_outline_prompt"]
+            + "Output language must in: "
+            + language,
         }
     )
     return {

src/apis/routers/gen_script.py

@@ -19,7 +19,7 @@ class GenScriptRequest(BaseModel):
     target_word_count: int = Field(
         2500, ge=2000, le=12000, description="Target word count"
     )
-
+    language: str = Field(..., description="Language")
 
 router = APIRouter()
 
@@ -147,6 +147,7 @@ async def gen_script(request: GenScriptRequest, user: user_dependency):
     input_graph = {
         "video_link": request.video_link,
         "target_word_count": request.target_word_count,
+        "language": request.language,
     }
 
     return StreamingResponse(