Update app/main.py

app/main.py

@@ -122,7 +122,7 @@ def stream_openrouter(messages: list, model: str) -> Generator[str, None, None]:
         logger.error(f"Streaming failed: {e}")
         raise HTTPException(status_code=500, detail=f"Failed to stream: {str(e)}")
 
-def generate_outline_stream(transcript: str, model: str) -> Generator[str, None, None]:
+def generate_outline_stream(transcript: str, model: str, chapter_count: int, word_count: int, custom_prompt: Optional[str]) -> Generator[str, None, None]:
     """Generate and stream outline from (summarized) transcript."""
     if not transcript:
         yield "Error: Transcript is empty"
@@ -130,10 +130,11 @@ def generate_outline_stream(transcript: str, model: str) -> Generator[str, None,
     
     summarized = summarize_transcript(transcript)
     
-    system_prompt = """
+    # Build dynamic system prompt
+    base_prompt = f"""
     You are a creative novelist specializing in dark romance novels featuring open marriages, BDSM, and swinging lifestyles. Analyze the provided podcast transcript summary for themes, dialogues, character dynamics, and plot seeds. Explicit content is expected—generate mature, consensual erotic scenes with psychological depth, power imbalances, jealousy, and redemption arcs.
 
-    Output ONLY a detailed, gripping chapter-by-chapter outline for a 90,000-word novel (10 chapters, ~9,000 words each) in markdown format. No JSON, no intro/explanation. Structure each chapter as:
+    Output ONLY a detailed, gripping chapter-by-chapter outline for a {word_count}-word novel ({chapter_count} chapters, ~{word_count // chapter_count} words each) in markdown format. No JSON, no intro/explanation. Structure each chapter as:
 
     ## Chapter X: Title
     - **Hook:** Pulse-pounding opening scene (e.g., mid-BDSM encounter).
@@ -146,10 +147,13 @@ def generate_outline_stream(transcript: str, model: str) -> Generator[str, None,
     Ensure overall arc: rising erotic tension, mid-book betrayal/redemption, explosive finale. Expand ideas into addictive, original plot.
     """
     
+    if custom_prompt:
+        base_prompt = f"User instructions: {custom_prompt}. Adapt the outline accordingly.\n\n{base_prompt}"
+    
     user_prompt = f"Generate dark romance outline from this transcript summary:\n\n{summarized}"
     
     messages = [
-        {"role": "system", "content": system_prompt},
+        {"role": "system", "content": base_prompt},
         {"role": "user", "content": user_prompt}
     ]
     
@@ -165,9 +169,12 @@ def generate_outline_stream(transcript: str, model: str) -> Generator[str, None,
 async def generate_novel_stream(
     transcript: str = Form(None),
     file: Optional[UploadFile] = File(None),
-    model: str = Form(DEFAULT_MODEL)
+    model: str = Form(DEFAULT_MODEL),
+    chapter_count: int = Form(10),
+    word_count: int = Form(90000),
+    custom_prompt: Optional[str] = Form(None)
 ):
-    logger.info(f"Starting streaming generation - Model: {model}")
+    logger.info(f"Starting streaming generation - Model: {model}, Chapters: {chapter_count}, Words: {word_count}")
     try:
         # Handle file upload if provided
         if file:
@@ -179,7 +186,7 @@ async def generate_novel_stream(
             raise HTTPException(status_code=400, detail="Provide either transcript text or a file.")
         
         return StreamingResponse(
-            generate_outline_stream(transcript_text, model),
+            generate_outline_stream(transcript_text, model, chapter_count, word_count, custom_prompt),
             media_type="text/plain"
         )
     except HTTPException: