Update app.py

app.py

@@ -74,15 +74,12 @@ import requests
 import os
 import gradio as gr
 
-# Fetch the API token from the environment variable set in Hugging Face Space Secrets
 API_TOKEN = os.getenv("HF_API_TOKEN")
 if not API_TOKEN:
     raise ValueError("HF_API_TOKEN not found. Please set it in the Space Secrets.")
 
-# DeepSeek-R1 model endpoint
 MODEL_URL = "https://api-inference.huggingface.co/models/deepseek-ai/DeepSeek-R1-Distill-Qwen-32B"
 
-# Headers for authentication
 HEADERS = {
     "Authorization": f"Bearer {API_TOKEN}",
     "Content-Type": "application/json"
@@ -93,17 +90,24 @@ def query_deepseek(prompt):
     payload = {
         "inputs": prompt,
         "parameters": {
-            "max_new_tokens": 100,  # Limit response length
-            "temperature": 0.6,     # Control creativity
-            "top_p": 0.9            # Nucleus sampling
+            "max_new_tokens": 100,
+            "temperature": 0.6,
+            "top_p": 0.9,
+            "repetition_penalty": 1.2  # Added to reduce repetition
         }
     }
     
     try:
         response = requests.post(MODEL_URL, headers=HEADERS, json=payload)
-        response.raise_for_status()  # Raise an error for bad status codes
+        response.raise_for_status()
         result = response.json()
-        return result[0]["generated_text"].strip()
+        
+        # Extract only the new response part
+        full_response = result[0]["generated_text"].strip()
+        if "Assistant: " in full_response:
+            # Split and get only the last assistant response
+            return full_response.split("Assistant: ")[-1].strip()
+        return full_response
     except requests.exceptions.RequestException as e:
         return f"Error: Could not connect to the API - {str(e)}"
     except (KeyError, IndexError):
@@ -115,6 +119,7 @@ def chat_interface(user_input, history):
     response = query_deepseek(prompt)
     return response
 
+# Keep your original CSS unchanged
 css = """
 @keyframes gradient {
     0% { background-position: 0% 50%; }
@@ -293,7 +298,6 @@ input[type="text"]:focus {
 }
 """
 
-# Create Gradio interface (no unsupported arguments)
 interface = gr.ChatInterface(
     fn=chat_interface,
     title="LocoBot: DeepSeek-R1 Chatbot",
@@ -303,7 +307,4 @@ interface = gr.ChatInterface(
 )
 
 if __name__ == "__main__":
-    interface.launch(server_name="0.0.0.0", server_port=7860)
-
-
-
+    interface.launch(server_name="0.0.0.0", server_port=7860)