Update app.py

app.py

@@ -192,6 +192,15 @@ class PodcastGenerator:
         
         add_log("⚠️ No valid JSON found, creating fallback")
         return self.create_fallback_podcast(text)
+        
+    def conversation_to_json(text: str) -> Dict:
+    """Convert speaker-formatted text to podcast JSON structure"""
+        lines = re.findall(r'(Speaker\s[12]):\s*(.+)', text)
+        podcast = [{"speaker": int(s[-1]), "line": l.strip()} for s, l in lines]
+        return {
+            "topic": "Generated from PDF",
+            "podcast": podcast
+        }
 
     def validate_podcast_structure(self, data: Dict) -> bool:
         """Validate podcast JSON structure"""
@@ -302,7 +311,7 @@ class PodcastGenerator:
         language_instruction = f"Generate in {language}" if language != "Auto Detect" else "Use appropriate language"
 
         # Simplified and more reliable prompt
-        system_prompt = f"""Create a podcast script in valid JSON format.
+        system_prompt = f"""Create a podcast script 
 
 Requirements:
 - Exactly 2 speakers (speaker 1 and 2)
@@ -311,7 +320,11 @@ Requirements:
 - DO NOT copy the example below , only use it as conversation reference 
 - The podcast should be professional, in-depth,  lively, witty and engaging, and hook the listener from the start.
 - The input text might be disorganized or unformatted. Ignore any formatting inconsistencies or irrelevant details; your task is to distill the essential points, identify key definitions, and highlight intriguing facts that would be suitable for discussion in a podcast.
-- The script must be in JSON format.
+- The two-person conversation must be in following format, always leading with the speaker ID 
+Speaker 1: Hello to our podcast
+Speaker 2: ...
+Speaker 1: ...
+Speaker 2: ...
 - {language_instruction}
 """
         
@@ -338,7 +351,7 @@ Requirements:
             )
             
             # Move to correct device
-            inputs = {k: v.to(self.model.device) for k, v in inputs.items()}
+            #inputs = {k: v.to(self.model.device) for k, v in inputs.items()}
             add_log(f"✅ Inputs moved to device: {self.model.device}")
 
             add_log("🧠 Generating with model...")
@@ -371,7 +384,8 @@ Requirements:
                 progress(0.4, "🔍 Processing generated script...")
 
             # Extract and validate JSON
-            result = self.clean_and_validate_json(generated_text)
+            #result = self.clean_and_validate_json(generated_text)
+            result = conversation_to_json(generated_text)
             
             if progress:
                 progress(0.5, "✅ Script generated successfully!")