Update app.py

app.py

@@ -3,130 +3,86 @@ from groq import Groq
 import gradio as gr
 import logging
 
-os.environ["GROQ_API_KEY"] = "c5417d24599246a4845e0aefba9a596f"
+os.environ["GROQ_API_KEY"] = "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"
 
-client = Groq(api_key=os.environ["GROQ_API_KEY"])
-
-# Set up logging configuration
 logging.basicConfig(level=logging.DEBUG)
 logger = logging.getLogger(__name__)
 
-# Initialize the Groq client with API key from environment variables
+# Initialize the Groq client
 api_key = os.environ.get("GROQ_API_KEY")
 if not api_key:
     logger.error("GROQ_API_KEY environment variable is not set.")
     raise ValueError("GROQ_API_KEY environment variable is required.")
 client = Groq(api_key=api_key)
 
-# Default model name, can be overridden by environment variable
 MODEL_NAME = os.environ.get("MODEL_NAME", "llama3-8b-8192")
 
+# Define a function to handle chat completions
 def get_completion(user_input):
-    """
-    Generate a chat completion response using the Groq client.
-
-    Args:
-        user_input (str): The user's input query.
-
-    Returns:
-        str: The generated response or an error message.
-    """
+    if not user_input.strip():
+        return "Please enter a valid query."
+    
+    # Check if the user asks "Who made you?"
+    if "who made you" in user_input.lower():
+        return "I was created by Thirumoorthi, a brilliant mind working on AI systems!"
+    
     try:
         completion = client.chat.completions.create(
             model=MODEL_NAME,
             messages=[
-                {"role": "system", "content": "You are a helpful assistant."},
+                {"role": "system", "content": "You are a friendly and helpful assistant, like ChatGPT."},
                 {"role": "user", "content": user_input}
             ],
-            temperature=1,
+            temperature=0.7,  # Slightly lower temperature for more controlled responses
             max_tokens=1024,
             top_p=1,
-            stream=False,  # Gradio does not support stream=True
+            stream=True,
+            stop=None,
         )
-
-        # Extract the response
-        response = completion.choices[0].message.content.strip()
-        return response
+        
+        response = ""
+        for chunk in completion:
+            response += chunk.choices[0].delta.content or ""
+        
+        return response.strip()  # Clean up response
     except Exception as e:
         logger.error(f"Error during completion: {e}")
         return "Sorry, I encountered an error while processing your request."
 
+# Launch Gradio interface
 def launch_interface():
-    """
-    Launch the Gradio interface for the chatbot.
-    """
     demo = gr.Interface(
         fn=get_completion,
         inputs=gr.Textbox(
-            label="Enter your query:",
-            placeholder="Ask me anything...",
+            label="Ask me anything:",
+            placeholder="I am here to help! Ask away...",
             lines=2,
-            max_lines=5
+            max_lines=5,
+            show_label=True,
+            interactive=True
         ),
         outputs=gr.Textbox(
             label="Response:",
+            interactive=False,
+            show_label=True,
             lines=6,
             max_lines=10
         ),
-        title="Mr AI",
-        description="""
-        <style>
-        .gr-box {
-            border-radius: 10px;
-            border: 2px solid #007BFF;
-            padding: 15px;
-            background-color: #F8F9FA;
-        }
-        .gr-input {
-            font-size: 1.2em;
-            padding: 10px;
-            border-radius: 5px;
-            border: 2px solid #007BFF;
-            margin-bottom: 10px;
-        }
-        .gr-output {
-            font-size: 1.2em;
-            padding: 10px;
-            border-radius: 5px;
-            border: 2px solid #28A745;
-            background-color: #E9F7EF;
-        }
-        .gr-interface-title {
-            font-size: 2em;
-            font-weight: bold;
-            color: #007BFF;
-            text-align: center;
-            margin-bottom: 20px;
-        }
-        .gr-interface-description {
-            font-size: 1.2em;
-            color: #6C757D;
-            text-align: center;
-            margin-bottom: 20px;
-        }
-        .gr-button {
-            background-color: #007BFF;
-            color: white;
-            border-radius: 5px;
-            padding: 10px 20px;
-            font-size: 1em;
-            border: none;
-            cursor: pointer;
-            margin-top: 10px;
-        }
-        .gr-button:hover {
-            background-color: #0056b3;
-        }
-        </style>
-        <div class="gr-interface-title">Welcome to Mr AI</div>
-        <div class="gr-interface-description">Ask anything and get a helpful response.</div>
+        title="Chat with Mr AI",
+        description="I am your friendly assistant, just like ChatGPT! Ask me anything, and I will do my best to help.",
+        theme="huggingface",  # More modern theme
+        css="""
+            .gr-box { border-radius: 15px; border: 1px solid #e1e1e1; padding: 20px; background-color: #f9f9f9; }
+            .gr-button { background-color: #4CAF50; color: white; font-size: 14px; }
+            .gr-textbox { border-radius: 8px; font-size: 16px; padding: 10px; }
+            .gr-output { background-color: #f1f1f1; border-radius: 8px; font-size: 16px; padding: 15px; }
         """,
         allow_flagging="never",
-        live=True
+        live=True,  # Enable live updates if supported
     )
-
+    
     logger.info("Starting Gradio interface")
     demo.launch(share=True)
 
 if __name__ == "__main__":
-    launch_interface()
+    launch_interface()