Update app.py

app.py

@@ -1,8 +1,9 @@
-#Codes from killerz3/PodGen & eswardivi/Podcastify
+#Using codes from killerz3/PodGen & eswardivi/Podcastify
 import json
 import spaces
 import httpx
 import os
+import re
 import asyncio
 import edge_tts
 import torch
@@ -146,7 +147,14 @@ def generator(messages):
     
     results = tokenizer.decode(output_ids[0][input_ids.shape[1]:], skip_special_tokens=True)
     return results
-    
+
+def extract_content(text):
+  """Extracts the JSON content from the given text."""
+  match = re.search(r'\{.*?\}', text, re.DOTALL)
+  if match:
+    return match.group(0)
+  else:
+    return None
 
 async def main(link):
     if not link.startswith("http://") and not link.startswith("https://"):
@@ -164,7 +172,7 @@ async def main(link):
         {"role": "user", "content": prompt},
     ]
     
-    generated_script = generator(messages)
+    generated_script = extract_content(generator(messages))
 
     print("Generated Script:"+generated_script)