Update app.py

app.py

@@ -7,7 +7,7 @@ client = InferenceClient("HuggingFaceH4/starchat2-15b-v0.1")
 
 def respond(
     message,
-    history: list[tuple[str, str]],
+    chat_history,
     system_message,
     max_tokens,
     temperature,
@@ -24,9 +24,9 @@ def respond(
     messages = [{"role": "system", "content": system_message}]
     
     # Build conversation history
-    for user_msg, assistant_msg in history:
-        if user_msg:
-            messages.append({"role": "user", "content": user_msg})
+    for human_msg, assistant_msg in chat_history:
+        if human_msg:
+            messages.append({"role": "user", "content": human_msg})
         if assistant_msg:
             messages.append({"role": "assistant", "content": assistant_msg})
     
@@ -43,9 +43,11 @@ def respond(
         ):
             token = message.choices[0].delta.content
             response += token
-            yield response
+            chat_history = chat_history + [(message, response)]
+            yield chat_history
     except Exception as e:
-        yield f"Error: {str(e)}"
+        chat_history = chat_history + [(message, f"Error: {str(e)}")]
+        yield chat_history
 
 def create_chat_interface():
     """
@@ -71,7 +73,8 @@ def create_chat_interface():
                     height=600,
                     show_label=False,
                     container=True,
-                    scale=2
+                    scale=2,
+                    type="messages"  # Set type to messages format
                 )
                 msg = gr.Textbox(
                     show_label=False,
@@ -116,17 +119,24 @@ def create_chat_interface():
                 with gr.Row():
                     clear = gr.Button("Clear Chat")
                     stop = gr.Button("Stop")
+
+        # Initialize chat history
+        state = gr.State([])
         
         # Handle sending messages
         msg.submit(
             respond,
-            [msg, chatbot, system_msg, max_tokens, temperature, top_p, model],
-            [chatbot],
+            [msg, state, system_msg, max_tokens, temperature, top_p, model],
+            [state],
             api_name="chat"
+        ).then(
+            lambda x: "", 
+            [msg], 
+            [msg]
         )
         
         # Clear chat history
-        clear.click(lambda: None, None, chatbot, queue=False)
+        clear.click(lambda: [], None, state, queue=False)
         
         # Example prompts
         gr.Examples(
@@ -144,4 +154,8 @@ def create_chat_interface():
 if __name__ == "__main__":
     demo = create_chat_interface()
     demo.queue()
-    demo.launch(share=True)
+    # Disable SSR and sharing for Spaces
+    demo.launch(
+        share=False,  # Disable sharing on Spaces
+        ssr=False    # Disable SSR to prevent warnings
+    )