Update app.py

app.py

@@ -75,7 +75,7 @@ def create_chat_app():
         temperature: float,
         top_p: float,
         language: str,
-    ) -> str:
+    ) -> Tuple[List[Tuple[str, str]], str]:
         try:
             client = Client("aifeifei798/feifei-chat")
             
@@ -99,10 +99,13 @@ def create_chat_app():
                 api_name="/chat"
             )
             
-            return response
+            chat_history.append((message, response))
+            return chat_history, ""  # Return updated history and empty message
             
         except Exception as e:
-            return TRANSLATIONS[language]["error_message"].format(str(e))
+            error_msg = TRANSLATIONS[language]["error_message"].format(str(e))
+            chat_history.append((message, error_msg))
+            return chat_history, ""
 
     def update_interface(language: str):
         trans = TRANSLATIONS[language]
@@ -133,7 +136,7 @@ def create_chat_app():
         with gr.Group():
             chatbot = gr.Chatbot(
                 value=[],
-                type="messages",  # Using the new messages format
+                height=400,
                 label=TRANSLATIONS["en"]["title"]
             )
             msg = gr.Textbox(
@@ -174,9 +177,8 @@ def create_chat_app():
         # Set up chat functionality
         msg.submit(
             respond,
-            [msg, chatbot, system_msg, max_tokens, temperature, top_p, language],
-            [chatbot],
-            clear_input=True
+            inputs=[msg, chatbot, system_msg, max_tokens, temperature, top_p, language],
+            outputs=[chatbot, msg]  # Update chat history and clear message input
         )
         
         # Examples