Update app.py

app.py

@@ -1,4 +1,4 @@
-import requests
+'''import requests
 import os
 import gradio as gr
 
@@ -66,5 +66,118 @@ interface = gr.ChatInterface(
     css=css
 )
 
+if __name__ == "__main__":
+    interface.launch(server_name="0.0.0.0", server_port=7860)'''
+
+import requests
+import os
+import gradio as gr
+
+# Fetch the API token from environment variable
+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"
+}
+
+def query_deepseek(prompt):
+    """Send a prompt to DeepSeek-R1 and get a response."""
+    payload = {
+        "inputs": prompt,
+        "parameters": {
+            "max_new_tokens": 100,  # Limit response length
+            "temperature": 0.6,     # Control creativity
+            "top_p": 0.9            # Nucleus sampling
+        }
+    }
+    
+    try:
+        response = requests.post(MODEL_URL, headers=HEADERS, json=payload)
+        response.raise_for_status()
+        result = response.json()
+        # Extract only the generated text, avoiding duplication
+        return result[0]["generated_text"].strip()
+    except requests.exceptions.RequestException as e:
+        return f"Error: Could not connect to the API - {str(e)}"
+    except (KeyError, IndexError):
+        return "Error: Unexpected response format from API"
+
+def chat_interface(user_input, history):
+    """Handle chatbot interaction with Gradio."""
+    # Construct the prompt with history if needed, but avoid repeating the response
+    if history:
+        prompt = "\n".join([f"User: {h[0]}\nAssistant: {h[1]}" for h in history]) + f"\nUser: {user_input}\nAssistant: "
+    else:
+        prompt = f"User: {user_input}\nAssistant: "
+    
+    response = query_deepseek(prompt)
+    # Ensure only the new response is returned, not the entire prompt
+    if response.startswith(prompt):
+        response = response[len(prompt):].strip()
+    return response
+
+# Custom CSS for an attractive and unique interface
+css = """
+body {
+    background: linear-gradient(135deg, #1e3c72, #2a5298); /* Gradient background */
+    font-family: 'Arial', sans-serif;
+}
+
+.gr-chat-interface {
+    border-radius: 15px;
+    box-shadow: 0 4px 20px rgba(0, 0, 0, 0.2);
+    background: rgba(255, 255, 255, 0.9);
+    padding: 20px;
+}
+
+button {
+    background-color: #007bff !important;  /* Vibrant blue buttons */
+    color: white !important;
+    border: none !important;
+    padding: 12px 24px !important;
+    border-radius: 25px !important;       /* Rounded buttons */
+    font-weight: bold;
+    transition: all 0.3s ease;
+}
+
+button:hover {
+    background-color: #0056b3 !important; /* Darker blue on hover */
+    transform: scale(1.05);               /* Slight zoom effect */
+}
+
+input, textarea {
+    border-radius: 10px !important;
+    border: 2px solid #007bff !important;
+    padding: 10px !important;
+}
+
+.title {
+    color: #ffffff;
+    text-shadow: 2px 2px 4px rgba(0, 0, 0, 0.3);
+    font-size: 2em;
+}
+
+.description {
+    color: #e0e0e0;
+    font-style: italic;
+}
+"""
+
+# Create Gradio interface
+interface = gr.ChatInterface(
+    fn=chat_interface,
+    title="LocoBot: DeepSeek-R1 Chatbot",
+    description="Chat with DeepSeek-R1 powered by Hugging Face Inference API.",
+    theme="default",
+    css=css
+)
+
 if __name__ == "__main__":
     interface.launch(server_name="0.0.0.0", server_port=7860)