Update app.py

Fixed interface.

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 import requests
-from functools import partial
 
 def send_message_with_secret(message, history, api_secret):
     if not api_secret.strip():
@@ -26,49 +25,6 @@ def send_message_with_secret(message, history, api_secret):
     except requests.exceptions.RequestException as e:
         return f"An error occurred: {e}"
 
-def create_chat_interface(api_secret):
-    # Partially apply the api_secret to the send_message function
-    send_fn = partial(send_message_with_secret, api_secret=api_secret)
-    
-    return gr.ChatInterface(
-        fn=send_fn,
-        title="Chat with Your Kin",
-        description="Send messages and receive responses in a chat-like interface",
-        examples=[
-            ["Hello! How are you today?"],
-            ["Tell me about yourself"],
-        ],
-        theme=gr.themes.Soft(
-            primary_hue="blue",
-            secondary_hue="gray",
-        ),
-        css="""
-            .message.user {
-                background-color: #2563eb !important;
-                color: white !important;
-                border-radius: 12px 12px 0 12px !important;
-                margin-left: auto !important;
-                margin-right: 1rem !important;
-            }
-            .message.bot {
-                background-color: #f3f4f6 !important;
-                border-radius: 12px 12px 12px 0 !important;
-                margin-right: auto !important;
-                margin-left: 1rem !important;
-            }
-            .message.user, .message.bot {
-                padding: 1rem !important;
-                margin-bottom: 1rem !important;
-                max-width: 80% !important;
-                display: inline-block !important;
-            }
-            .message-wrap {
-                display: flex !important;
-                flex-direction: column !important;
-            }
-        """
-    )
-
 with gr.Blocks() as app:
     gr.Markdown("## AI Chat Interface")
     
@@ -79,15 +35,38 @@ with gr.Blocks() as app:
         show_label=True
     )
     
-    chat_bot = gr.Chatbot(height=500)
+    chatbot = gr.Chatbot(
+        value=[],
+        type='messages',
+        height=400
+    )
+    
+    msg = gr.Textbox(
+        label="Message",
+        placeholder="Type your message here...",
+        show_label=False
+    )
+    
+    def respond(message, chat_history, api_secret):
+        if not message:
+            return chat_history, ""
+            
+        bot_response = send_message_with_secret(message, chat_history, api_secret)
+        chat_history.append({"role": "user", "content": message})
+        chat_history.append({"role": "assistant", "content": bot_response})
+        return chat_history, ""
     
-    def update_chat(api_key):
-        return create_chat_interface(api_key)
+    msg.submit(
+        respond,
+        inputs=[msg, chatbot, api_secret],
+        outputs=[chatbot, msg]
+    )
     
-    api_secret.change(
-        fn=update_chat,
-        inputs=[api_secret],
-        outputs=[chat_bot]
+    send_btn = gr.Button("Send")
+    send_btn.click(
+        respond,
+        inputs=[msg, chatbot, api_secret],
+        outputs=[chatbot, msg]
     )
 
 if __name__ == "__main__":