Update app.py

app.py

@@ -1,26 +1,27 @@
 import gradio as gr
 import os
 import requests
-
-# Define the Hugging Face API details
-API_URL = "https://api-inference.huggingface.co/models/Huzaifa367/chat-summarizer"
-API_TOKEN = os.getenv("AUTH_TOKEN")
-HEADERS = {"Authorization": f"Bearer {API_TOKEN}"}
-
-def predict(message, history):
+def query_huggingface(payload):
     try:
-        # Send request to the Hugging Face API with only the current message
-        payload = {"inputs": message}
         response = requests.post(API_URL, headers=HEADERS, json=payload)
         response.raise_for_status()  # Raise exception for non-2xx status codes
-
-        # Extract summary text from the response
-        summary_text = response.json()["summary_text"]
+        return response.json()
     except requests.exceptions.RequestException as e:
-        # Handle API request errors
-        summary_text = f"Error querying Hugging Face API: {e}"
-    
+        print(f"Error querying Hugging Face API: {e}")
+        return {"summary_text": f"Error querying Hugging Face API: {e}"}
+
+def respond(user_message):
+    # Construct input text for summarization (only user message)
+    input_text = f"User: {user_message}"
+
+    # Query Hugging Face API for summarization
+    payload = {"inputs": input_text}
+    response = query_huggingface(payload)
+
+    # Extract summary text from the API response
+    summary_text = response.get("summary_text", "No response from Hugging Face API")
+
     return summary_text
 
 # Launch Gradio chat interface with the predict function
-gr.ChatInterface(predict).launch()
+gr.ChatInterface(respond).launch()