Update app.py

app.py

@@ -87,10 +87,10 @@ def chat_model(message, history, temperature=0.7, top_p=0.9):
     
     # Format chat history for the model
     formatted_history = []
-    for human_msg, ai_msg in history:
-        formatted_history.append({"role": "user", "content": human_msg})
-        if ai_msg:  # Skip None values that might occur during streaming
-            formatted_history.append({"role": "assistant", "content": ai_msg})
+    for h in history:
+        formatted_history.append({"role": "user", "content": h["content"] if h["role"] == "user" else ""})
+        if h["role"] == "assistant":
+            formatted_history.append({"role": "assistant", "content": h["content"]})
     
     # Add the current message
     formatted_history.append({"role": "user", "content": message})
@@ -217,31 +217,37 @@ css = """
 with gr.Blocks(theme=theme, title="Stanford Soft Raccoon Chat", css=css) as demo:
     with gr.Row():
         with gr.Column(scale=3):
-            # Use ChatInterface for the main chat functionality
-            chatbot = gr.ChatInterface(
+            # Create the chatbot component directly
+            chatbot = gr.Chatbot(
+                label="Soft Raccoon Chat",
+                avatar_images=(None, "🌲"),  # Stanford tree emoji
+                height=600,
+                placeholder="<strong>Soft Raccoon AI Assistant</strong><br>Ask me anything!"
+            )
+            
+            # Create sliders for temperature and top_p
+            with gr.Accordion("Generation Parameters", open=False):
+                temperature = gr.Slider(
+                    minimum=0.1,
+                    maximum=2.0,
+                    value=0.7,
+                    step=0.1,
+                    label="Temperature"
+                )
+                top_p = gr.Slider(
+                    minimum=0.1,
+                    maximum=1.0,
+                    value=0.9,
+                    step=0.05,
+                    label="Top-P"
+                )
+            
+            # Create the ChatInterface
+            chat_interface = gr.ChatInterface(
                 fn=chat_model,
-                chatbot=gr.Chatbot(
-                    label="Soft Raccoon Chat",
-                    avatar_images=(None, "🌲"),  # Stanford tree emoji
-                    height=600,
-                    placeholder="<strong>Soft Raccoon AI Assistant</strong><br>Ask me anything!"
-                ),
-                additional_inputs=[
-                    gr.Slider(
-                        minimum=0.1,
-                        maximum=2.0,
-                        value=0.7,
-                        step=0.1,
-                        label="Temperature"
-                    ),
-                    gr.Slider(
-                        minimum=0.1,
-                        maximum=1.0,
-                        value=0.9,
-                        step=0.05,
-                        label="Top-P"
-                    )
-                ],
+                chatbot=chatbot,
+                additional_inputs=[temperature, top_p],
+                type="messages",  # This is important for compatibility
                 title="Stanford Soft Raccoon Chat",
                 description="AI assistant powered by the Soft Raccoon language model",
                 examples=[
@@ -250,9 +256,6 @@ with gr.Blocks(theme=theme, title="Stanford Soft Raccoon Chat", css=css) as demo
                     "What's your favorite book?"
                 ],
                 cache_examples=True,
-                retry_btn="Regenerate",
-                undo_btn="Undo",
-                clear_btn="Clear",
             )
         
         with gr.Column(scale=1):