Update app.py

app.py

@@ -88,13 +88,14 @@ HEADERS = {
 }
 
 def query_deepseek(prompt):
-    """Send a prompt to DeepSeek-R1 and get a response."""
+    """Send a prompt to DeepSeek-R1 and get a clean response."""
     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": 80,    # Reduced to limit repetition
+            "temperature": 0.5,      # Lowered for less randomness
+            "top_p": 0.85,           # Tighter nucleus sampling
+            "stop": ["\n", "User:", "Assistant:"]  # Stop at natural breaks
         }
     }
     
@@ -103,6 +104,9 @@ def query_deepseek(prompt):
         response.raise_for_status()
         result = response.json()
         generated_text = result[0]["generated_text"].strip()
+        # Remove the prompt if it’s echoed back
+        if prompt in generated_text:
+            return generated_text.replace(prompt, "").strip()
         return generated_text
     except requests.exceptions.RequestException as e:
         return f"Error: Could not connect to the API - {str(e)}"
@@ -111,87 +115,120 @@ def query_deepseek(prompt):
 
 def chat_interface(user_input, history):
     """Handle chatbot interaction with Gradio."""
-    # Construct a clean prompt
+    # Build a concise prompt with history
     if history:
-        prompt = "\n".join([f"User: {h[0]}\nAssistant: {h[1]}" for h in history]) + f"\nUser: {user_input}\nAssistant: "
+        prompt = "\n".join([f"U: {h[0]} | A: {h[1]}" for h in history]) + f"\nU: {user_input} | A: "
     else:
-        prompt = f"User: {user_input}\nAssistant: "
+        prompt = f"U: {user_input} | A: "
     
-    # Get the response
+    # Get response and ensure no duplication
     response = query_deepseek(prompt)
     
-    # Strip the prompt to avoid duplication
-    if response.startswith(prompt):
-        response = response[len(prompt):].strip()
+    # Clean up any lingering prompt markers
+    response = response.replace("U: ", "").replace("| A: ", "").strip()
     
-    # Deduplicate sentences
+    # Final deduplication: split into sentences and keep unique ones
     sentences = response.split(". ")
     unique_sentences = []
-    [unique_sentences.append(s) for s in sentences if s not in unique_sentences]
-    response = ". ".join(unique_sentences).strip()
-    
-    return response
+    [unique_sentences.append(s) for s in sentences if s not in unique_sentences and s]
+    return ". ".join(unique_sentences).strip()
 
-# Simplified black-and-white CSS with a blue send button
+# New interactive CSS with a futuristic, playful vibe
 css = """
 body {
-    background: white;  /* Plain white background */
-    color: black;       /* Black text */
-    font-family: 'Arial', sans-serif;
+    background: #000000;  /* Deep black background */
+    color: #00ffcc;       /* Neon cyan text */
+    font-family: 'Courier New', monospace;  /* Retro terminal feel */
 }
 
 .gr-chat-interface {
-    background: white;
-    color: black;
+    background: rgba(0, 0, 0, 0.8);
+    border: 2px solid #00ffcc;
+    border-radius: 10px;
+    padding: 20px;
+    max-width: 700px;
+    margin: 20px auto;
 }
 
-button#submit_btn {  /* Target the send button specifically */
-    background-color: #007bff !important;  /* Blue send button */
-    color: white !important;
-    border: none !important;
-    padding: 10px 20px !important;
-    border-radius: 5px !important;
+.chatbot-container {
+    height: 400px;
+    overflow-y: auto;
+    background: #1a1a1a;
+    border-radius: 8px;
+    padding: 10px;
+    color: #00ffcc;
 }
 
-button#submit_btn:hover {
-    background-color: #0056b3 !important; /* Darker blue on hover */
+/* Style the chat messages */
+.chatbot-container .message {
+    margin: 10px 0;
+    padding: 8px;
+    border-radius: 5px;
+    background: #333333;
+    transition: transform 0.2s ease;
 }
 
-button {  /* Other buttons remain black and white */
-    background-color: white !important;
-    color: black !important;
-    border: 1px solid black !important;
-    padding: 10px 20px !important;
-    border-radius: 5px !important;
+.chatbot-container .message:hover {
+    transform: translateX(5px);  /* Slight slide on hover */
 }
 
-input, textarea {
-    background: white !important;
-    color: black !important;
-    border: 1px solid black !important;
+input {
+    background: #1a1a1a !important;
+    color: #00ffcc !important;
+    border: 1px solid #00ffcc !important;
     border-radius: 5px !important;
     padding: 10px !important;
+    font-family: 'Courier New', monospace;
+}
+
+button#submit_btn {
+    background: #ff007a !important;  /* Neon pink send button */
+    color: #ffffff !important;
+    border: none !important;
+    padding: 10px 20px !important;
+    border-radius: 20px !important;
+    font-family: 'Courier New', monospace;
+    font-weight: bold;
+    box-shadow: 0 0 10px #ff007a;
+    transition: all 0.3s ease;
+}
+
+button#submit_btn:hover {
+    background: #cc0066 !important;  /* Darker pink on hover */
+    box-shadow: 0 0 15px #ff007a;
+    transform: scale(1.1);  /* Slight grow effect */
 }
 
 .title {
-    color: black;
-    font-size: 1.5em;
+    color: #ff007a;
+    font-size: 2em;
+    text-align: center;
+    text-transform: uppercase;
+    letter-spacing: 2px;
+    animation: glow 1.5s infinite alternate;
 }
 
 .description {
-    color: black;
-    font-style: italic;
+    color: #00ffcc;
+    text-align: center;
+    font-size: 0.9em;
+}
+
+/* Glow animation for title */
+@keyframes glow {
+    from { text-shadow: 0 0 5px #ff007a; }
+    to { text-shadow: 0 0 15px #ff007a, 0 0 25px #ff007a; }
 }
 """
 
-# Create Gradio interface
+# Create Gradio interface with custom chatbot
 interface = gr.ChatInterface(
     fn=chat_interface,
-    title="LocoBot: DeepSeek-R1 Chatbot",
-    description="Chat with DeepSeek-R1 powered by Hugging Face Inference API.",
-    theme="default",
+    title="LocoBot: Neural Nexus",
+    description="A quirky AI companion powered by DeepSeek-R1. Ask away!",
     css=css,
-    submit_btn=gr.Button("Send", elem_id="submit_btn")  # Custom send button with ID
+    submit_btn=gr.Button("Transmit", elem_id="submit_btn"),
+    textbox=gr.Textbox(placeholder="Type your query, human..."),
 )
 
 if __name__ == "__main__":