Update app.py

app.py

@@ -1,6 +1,6 @@
 import requests
 import os
-import time
+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")
@@ -37,27 +37,36 @@ def query_deepseek(prompt):
     except (KeyError, IndexError):
         return "Error: Unexpected response format from API"
 
-def chatbot():
-    """Simple chatbot loop for command-line interaction."""
-    print("Welcome to the DeepSeek-R1 Chatbot! Type 'exit' to quit.")
-    while True:
-        user_input = input("You: ")
-        if user_input.lower() == "exit":
-            print("Goodbye!")
-            break
-        
-        # Structure the prompt
-        prompt = f"User: {user_input}\nAssistant: "
-        print("Thinking...", end="\r")
-        
-        # Get response from DeepSeek-R1
-        response = query_deepseek(prompt)
-        
-        # Clear "Thinking..." and display response
-        print(" " * 20, end="\r")
-        print(f"Bot: {response}")
-        
-        time.sleep(1)
+def chat_interface(user_input, history):
+    """Handle chatbot interaction with Gradio."""
+    prompt = f"User: {user_input}\nAssistant: "
+    response = query_deepseek(prompt)
+    return response
+
+# Custom CSS for blue buttons
+css = """
+button {
+    background-color: #007bff !important;  /* Blue background */
+    color: white !important;              /* White text */
+    border: none !important;
+    padding: 10px 20px !important;
+    border-radius: 5px !important;
+}
+button:hover {
+    background-color: #0056b3 !important; /* Darker blue on hover */
+}
+"""
+
+# 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,
+    submit_btn="Send",
+    clear_btn="Clear Chat"
+)
 
 if __name__ == "__main__":
-    chatbot()
+    interface.launch(server_name="0.0.0.0", server_port=7860)