Update app.py

app.py

@@ -48,25 +48,23 @@ def messages_to_history(messages: Messages) -> Tuple[str, History]:
     return system, history
 
 # Main function for chat
-import requests
-
 def model_chat(query: Optional[str], history: Optional[History], system: str) -> Tuple[str, str, History]:
     if query is None:
         query = ''
     if history is None:
         history = []
     
-    # Convert history to a string
+    # Convert history to a list of messages
     messages = history_to_messages(history, system)
     messages.append({'role': 'user', 'content': query})
     
     # Prepare the payload for Hugging Face Inference API
     payload = {"inputs": query, "parameters": {"max_new_tokens": 150}, "history": messages}
-    headers = {"Authorization": f"Bearer {os.getenv('HF_TOKEN')}"}
+    headers = {"Authorization": f"Bearer {os.getenv('YOUR_API_TOKEN')}"}
     
     try:
         # Request generation with Hugging Face Inference API
-        response = requests.post(f"https://api-inference.huggingface.co/models/Qwen/Qwen2.5-72B-Instruct", 
+        response = requests.post(f"https://api-inference.huggingface.co/models/YourModelNameHere", 
                                  json=payload, headers=headers)
         
         if response.status_code == 200:
@@ -81,19 +79,20 @@ def model_chat(query: Optional[str], history: Optional[History], system: str) ->
             # Log the chat to file
             log_history_to_file(query, response_text)
             
-            # Update history and return response
+            # Update history with the new assistant response and return it
+            history.append([query, response_text])
             system, history = messages_to_history(messages + [{'role': 'assistant', 'content': response_text}])
-            yield '', history, system
+            return response_text, history, system
         else:
             # Log error message to file and display it in Gradio
             error_message = f"Error {response.status_code}: {response.json().get('error', response.text)}"
             log_history_to_file(query, error_message)
-            yield error_message, history, system
+            return error_message, history, system
     except Exception as e:
         # Log any unexpected exceptions
         error_message = f"Exception: {str(e)}"
         log_history_to_file(query, error_message)
-        yield error_message, history, system
+        return error_message, history, system
 
 
 # Gradio Interface Setup