Update app.py

app.py

@@ -5,7 +5,6 @@ import logging
 
 os.environ["GROQ_API_KEY"] = "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"
 
-
 logging.basicConfig(level=logging.DEBUG)
 logger = logging.getLogger(__name__)
 
@@ -20,6 +19,9 @@ MODEL_NAME = os.environ.get("MODEL_NAME", "llama3-8b-8192")
 
 # Define a function to handle chat completions
 def get_completion(user_input):
+    if not user_input.strip():
+        return "Please enter a valid query."
+    
     try:
         completion = client.chat.completions.create(
             model=MODEL_NAME,
@@ -38,7 +40,7 @@ def get_completion(user_input):
         for chunk in completion:
             response += chunk.choices[0].delta.content or ""
         
-        return response
+        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."
@@ -65,8 +67,13 @@ def launch_interface():
         title="Mr AI",
         description="Ask anything and get a helpful response.",
         theme="default",
-        css=".gr-box { border-radius: 10px; border: 1px solid #ccc; padding: 10px; }",
-        allow_flagging="never"
+        css="""
+            .gr-box { border-radius: 10px; border: 1px solid #ccc; padding: 10px; }
+            .gr-button { background-color: #4CAF50; color: white; }
+            .gr-textbox { border-radius: 5px; }
+        """,
+        allow_flagging="never",
+        live=True,  # Enable live updates if supported
     )
     
     logger.info("Starting Gradio interface")