Upload 3 files

app.py

@@ -237,12 +237,12 @@ with gr.Blocks(title="Context Window Extender - Chat") as demo:
     ):
         """Handle chat response with streaming."""
         if not message.strip():
-            yield [(user_msg, "Please enter a message.") for user_msg, _ in history] + [(message, "Please enter a message.")]
+            yield [{"role": "user", "content": msg} for msg, _ in history] + [{"role": "user", "content": message, "content": "Please enter a message."}]
             return
         
         # Add user message to history
-        history.append((message, ""))
-        yield history
+        history.append({"role": "user", "content": message})
+        yield history + [{"role": "assistant", "content": "..."}]
         
         # Generate response
         try:
@@ -251,8 +251,10 @@ with gr.Blocks(title="Context Window Extender - Chat") as demo:
             
             # Build prompt from history
             prompt = message
-            for user_msg, assistant_msg in history[:-1]:
-                prompt = f"User: {user_msg}\nAssistant: {assistant_msg}\n" + prompt
+            for item in history[:-1]:
+                role = item.get("role", "user")
+                content = item.get("content", "")
+                prompt = f"User: {content}\nAssistant: " + prompt
             
             prompt = prompt + "\nAssistant:"
             
@@ -292,20 +294,18 @@ with gr.Blocks(title="Context Window Extender - Chat") as demo:
             for text in streamer:
                 full_response += text
                 # Update the last message (assistant response)
-                history[-1] = (message, full_response)
-                yield history
+                current_history = history + [{"role": "assistant", "content": full_response}]
+                yield current_history
             
             thread.join()
             
             if not full_response.strip():
                 full_response = "Model generated same text as input. Try adjusting parameters."
-                history[-1] = (message, full_response)
-                yield history
+                yield history + [{"role": "assistant", "content": full_response}]
             
         except Exception as e:
             full_response = f"Error: {str(e)}"
-            history[-1] = (message, full_response)
-            yield history
+            yield history + [{"role": "assistant", "content": full_response}]
     
     # ChatInterface
     chat_interface = gr.ChatInterface(