Revert to chatInterface

app.py

@@ -150,40 +150,18 @@ with gr.Blocks(css=fancy_css) as demo:
         type="messages",
         avatar_images=(str(ASSETS_DIR / "monster_icon.png"), str(ASSETS_DIR / "smart_confidant_icon.png"))
     )
-    
-    # Message input textbox
-    msg = gr.Textbox(placeholder="Type your message here...", label="Message")
-    
-    # Additional inputs in accordion below chatbot
-    with gr.Accordion("Advanced Settings", open=False):
-        system_message = gr.Textbox(value=DEFAULT_SYSTEM_MESSAGE, label="System message")
-        max_tokens = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens")
-        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.95, step=0.05, label="Top-p (nucleus sampling)")
-        selected_model = gr.Radio(choices=MODEL_OPTIONS, label="Select Model", value=MODEL_OPTIONS[1])
-    
-    # Handle message submission  
-    def handle_message(message, history, sys_msg, max_tok, temp, top_p_val, model):
-        if message.strip():
-            # For this simplified version, we'll use None for hf_token
-            # The respond function will handle the case where token is None
-            hf_token = None
-            
-            # Add user message to history
-            history.append({"role": "user", "content": message})
-            # Generate response
-            response_gen = respond(message, history[:-1], sys_msg, max_tok, temp, top_p_val, hf_token, model)
-            response = ""
-            for partial_response in response_gen:
-                response = partial_response
-            # Add assistant response to history
-            history.append({"role": "assistant", "content": response})
-        return "", history
-    
-    msg.submit(
-        handle_message,
-        inputs=[msg, chatbot, system_message, max_tokens, temperature, top_p, selected_model],
-        outputs=[msg, chatbot]
+    # Create ChatInterface with the custom chatbot and additional inputs in accordion below
+    gr.ChatInterface(
+        fn=respond,
+        chatbot=chatbot,
+        additional_inputs=[
+            gr.Textbox(value=DEFAULT_SYSTEM_MESSAGE, label="System message"),
+            gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
+            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.95, step=0.05, label="Top-p (nucleus sampling)"),
+            gr.Radio(choices=MODEL_OPTIONS, label="Select Model", value=MODEL_OPTIONS[2]),
+        ],
+        type="messages",
     )
 
 if __name__ == "__main__":