Update app.py

app.py

@@ -82,37 +82,50 @@ client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
 def respond(
     message,
-    history: list[tuple[str, str]],
+    chat_history,
     system_message,
     max_tokens,
     temperature,
     top_p,
 ):
+    if message.strip() == "":
+        return "", chat_history
+    
     messages = [{"role": "system", "content": system_message}]
-    for val in history:
-        if val[0]:
-            messages.append({"role": "user", "content": val[0]})
-        if val[1]:
-            messages.append({"role": "assistant", "content": val[1]})
+    
+    # Convert chat history to the format expected by the model
+    for human, assistant in chat_history:
+        messages.append({"role": "user", "content": human})
+        if assistant:  # Only add assistant message if it exists
+            messages.append({"role": "assistant", "content": assistant})
+    
     messages.append({"role": "user", "content": message})
-    response = ""
-    for message in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        token = message.choices[0].delta.content
-        response += token
-        yield response
+    
+    # Get the response from the model
+    try:
+        response = ""
+        for chunk in client.chat_completion(
+            messages,
+            max_tokens=max_tokens,
+            stream=True,
+            temperature=temperature,
+            top_p=top_p,
+        ):
+            if hasattr(chunk.choices[0].delta, 'content'):
+                if chunk.choices[0].delta.content is not None:
+                    response += chunk.choices[0].delta.content
+                    
+        chat_history.append((message, response))
+        return "", chat_history
+    except Exception as e:
+        return "", chat_history + [(message, f"Error: {str(e)}")]
 
 with gr.Blocks(css=custom_css) as demo:
     with gr.Column(elem_id="chat-container"):
         # Header
         with gr.Row(elem_id="header"):
             with gr.Column(elem_id="logo"):
-                gr.Markdown("🔄 DeepAI")
+                gr.Markdown("🌎 ZiF-V0")
             with gr.Row(elem_id="nav-links"):
                 gr.Markdown("[AI Chat](#) [AI Image Generator](#) [AI Video](#) [AI Music Generator](#) [Login](#)")
         
@@ -164,9 +177,13 @@ with gr.Blocks(css=custom_css) as demo:
             # Set up chat functionality
             msg.submit(
                 respond,
-                [msg, chatbot, system_message, max_tokens, temperature, top_p],
-                [msg, chatbot]
+                inputs=[msg, chatbot, system_message, max_tokens, temperature, top_p],
+                outputs=[msg, chatbot]
             )
 
+            # Add a clear button
+            clear = gr.Button("Clear")
+            clear.click(lambda: None, None, chatbot, queue=False)
+
 if __name__ == "__main__":
     demo.launch()