Update app.py

app.py

@@ -66,12 +66,12 @@ def format_conversation(history, system_prompt):
     """
     prompt = system_prompt.strip() + "\n"
     for msg in history:
-        if msg['role'] == 'user':
-            prompt += "User: " + msg['content'].strip() + "\n"
-        elif msg['role'] == 'assistant':
-            prompt += "Assistant: " + msg['content'].strip() + "\n"
+        if msg["role"] == "user":
+            prompt += "User: " + msg["content"].strip() + "\n"
+        elif msg["role"] == "assistant":
+            prompt += "Assistant: " + msg["content"].strip() + "\n"
         else:
-            prompt += msg['content'].strip() + "\n"
+            prompt += msg["content"].strip() + "\n"
     if not prompt.strip().endswith("Assistant:"):
         prompt += "Assistant: "
     return prompt
@@ -85,10 +85,10 @@ def chat_response(user_msg, chat_history, system_prompt,
     """
     cancel_event.clear()
     history = list(chat_history or [])
-    history.append({'role': 'user', 'content': user_msg})
+    history.append({"role": "user", "content": user_msg})
 
     # Prepare assistant placeholder
-    history.append({'role': 'assistant', 'content': ''})
+    history.append({"role": "assistant", "content": ""})
 
     try:
         prompt = format_conversation(history, system_prompt)
@@ -117,11 +117,11 @@ def chat_response(user_msg, chat_history, system_prompt,
             if cancel_event.is_set():
                 break
             assistant_text += chunk
-            history[-1]['content'] = assistant_text
+            history[-1]["content"] = assistant_text
             yield history
         gen_thread.join()
     except Exception as e:
-        history[-1]['content'] = f"Error: {e}"
+        history[-1]["content"] = f"Error: {e}"
         yield history
     finally:
         gc.collect()