Update app.py

app.py

@@ -15,7 +15,6 @@ def level_to_prompt(level):
 # Custom background CSS
 css = """
 @import url('https://fonts.googleapis.com/css2?family=Noto+Sans+JP&family=Playfair+Display&display=swap');
-
 body {
   background-image: url('https://cdn-uploads.huggingface.co/production/uploads/67351c643fe51cb1aa28f2e5/wuyd5UYTh9jPrMJGmV9yC.jpeg');
   background-size: cover;
@@ -58,25 +57,38 @@ def respond(message, history, level, max_tokens, temperature, top_p):
     system_message = level_to_prompt(level)
     messages = [{"role": "system", "content": system_message}]
     
-    for user, bot in history:
-        if user:
-            messages.append({"role": "user", "content": user})
-        if bot:
-            messages.append({"role": "assistant", "content": bot})
+    # Handle history based on its format
+    if history and isinstance(history[0], dict):
+        # New format (messages with role/content)
+        messages.extend(history)
+    else:
+        # Old format (tuples)
+        for user, bot in history:
+            if user:
+                messages.append({"role": "user", "content": user})
+            if bot:
+                messages.append({"role": "assistant", "content": bot})
+    
+    # Add current message
     messages.append({"role": "user", "content": message})
-
+    
+    # Generate response
     response = ""
-    for msg in client.chat_completion(
-        messages, max_tokens=max_tokens, stream=True, temperature=temperature, top_p=top_p
-    ):
-        token = msg.choices[0].delta.content
-        response += token
-        yield response
+    try:
+        for msg in client.chat_completion(
+            messages, max_tokens=max_tokens, stream=True, temperature=temperature, top_p=top_p
+        ):
+            token = msg.choices[0].delta.content
+            if token is not None:  # Handle None tokens
+                response += token
+            yield response
+    except Exception as e:
+        print(f"Error in chat completion: {e}")
+        yield f"Désolé! There was an error: {str(e)}"
 
 # UI layout
 with gr.Blocks(css=css) as demo:
     gr.Markdown("French Tutor", elem_id="custom-title")
-
     with gr.Column(elem_id="chat-panel"):
         with gr.Accordion("⚙️ Advanced Settings", open=False):
             level = gr.Dropdown(
@@ -95,4 +107,4 @@ with gr.Blocks(css=css) as demo:
         )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()