Update app.py

app.py

@@ -1,49 +1,43 @@
-import os
 import gradio as gr
 from groq import Groq
 import traceback
 import sys
 
-# Get GROQ API Key from environment
-api_key = os.environ.get("GROQ_API_KEY")
-if not api_key:
-    print("❌ GROQ_API_KEY is missing.", file=sys.stderr)
-
-client = Groq(api_key=api_key)
-
-# Chat function with logging to stderr (for Spaces)
-def chat_inference(message, history, example_prompt):
+# Chat function accepts API key as user input
+def chat_inference(message, history, api_key, example_prompt):
     try:
-        print(f"📥 User message: {message}", file=sys.stderr)
+        print(f"📥 Message: {message}", file=sys.stderr)
 
+        # Strip whitespace/newlines
+        api_key = api_key.strip()
+        client = Groq(api_key=api_key)
+
+        # Call Groq API
         response = client.chat.completions.create(
             messages=[{"role": "user", "content": message}],
             model="compound-beta"
         )
 
         reply = response.choices[0].message.content
-        print(f"📤 Groq reply: {reply}", file=sys.stderr)
+        print(f"📤 Reply: {reply}", file=sys.stderr)
 
     except Exception as e:
-        error_trace = traceback.format_exc()
-        print("❌ Exception occurred:", file=sys.stderr)
-        print(error_trace, file=sys.stderr)
-
-        reply = "⚠️ Error: Connection error or API issue.\n\nCheck the Logs tab in Hugging Face Spaces for more details."
+        print(traceback.format_exc(), file=sys.stderr)
+        reply = "⚠️ Error: Connection error or invalid API key. See logs in Spaces tab."
     return reply
 
-# Optional textbox for additional input
+# Inputs: API key + optional example prompt
 additional_inputs = [
+    gr.Textbox(label="🔐 Groq API Key", placeholder="Paste your API key here", type="password"),
     gr.Textbox(label="🔍 Example Prompt", value="What were the main highlights from the latest Apple keynote?")
 ]
 
-# Gradio interface
 chat_interface = gr.ChatInterface(
     fn=chat_inference,
     additional_inputs=additional_inputs,
     additional_inputs_accordion=gr.Accordion("⚙️ Configuration & Advanced Parameters", open=True),
     title="🔍 AI-Powered Real-Time Search with Groq",
-    description="Ask anything that requires real-time info — powered by Groq’s `compound-beta` model.",
+    description="Ask questions using Groq’s `compound-beta` model with real-time web access.",
     theme="default",
     type="messages",
 )