Update app.py

app.py

@@ -166,14 +166,14 @@ def openai_chat_avatar(text_prompt):
         model="gpt-4o-mini",
         messages=[
             {"role": "system", "content": "Summarize the following paragraph into a complete and accurate single sentence with no more than 15 words. The summary should capture the gist of the paragraph and make sense."},
-            {"role": "user", "content": f"Please summarize the following paragraph into one sentence with 15 words or fewer, ensuring it makes sense and captures the gist: {text_prompt}"},
+            {"role": "user", "content": f"Please summarize the following paragraph into one sentence with 15 words or fewer, ensuring it makes sense and captures the gist and remove the citation from the end: {text_prompt}"},
         ],
         max_tokens = len(text_prompt),  # Limit the response to a reasonable length for a summary
     )
     return response
 
 def ryzedb_chat_avatar(question, app_id):
-    url = "https://inference.dev.ryzeai.ai/chat/stream"
+    url = "https://inference.dev.ryzeai.ai/v2/chat/stream"
     # question = question + ". Summarize the answer in one line."
     # print("question",question)
     payload = json.dumps({
@@ -182,7 +182,8 @@ def ryzedb_chat_avatar(question, app_id):
     "app_id": app_id,
     "question": question
     },
-    "config": {}
+    "config": {
+        "thread_id": "123456"
     })
     headers = {
         'Content-Type': 'application/json'
@@ -289,13 +290,18 @@ def run_preprocessing(args):
 #     cleaned_text = re.sub(r'\s*\[.*?\]\s*$', '', text)
 #     return cleaned_text.strip()
 
-def extract_content(data):
-    pattern = r'"content":"((?:\\.|[^"\\])*)"'
-    match = re.search(pattern, data)
-    if match:
-        return match.group(1)
-    else:
-        return None
+def extract_content(line):
+    print("line :", line)
+    if line.startswith("data:"):
+        print("Entering extract content")
+        json_part = line[len("data:"):].strip()
+        if json_part:
+            try:
+                parsed = json.loads(json_part)
+                return parsed.get("content")
+            except json.JSONDecodeError:
+                pass
+    return None
 
 @app.route("/run", methods=['POST'])
 def generate_video():