Update app.py

app.py

@@ -2,17 +2,20 @@ import os
 import gradio as gr
 from groq import Groq
 
-# ✅ API KEY
+# ✅ Load GROQ API Key from Hugging Face Secrets
 GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
 if not GROQ_API_KEY:
-    raise ValueError("❌ Please add GROQ_API_KEY in your Hugging Face Space Secrets!")
+    raise ValueError("❌ Please add your GROQ_API_KEY in Hugging Face Space Secrets!")
 
+# ✅ Groq model
 MODEL_NAME = "llama3-8b-8192"
 
+# ✅ System prompt
 SYSTEM_PROMPT = """
 You are Dr Cat, a friendly cat expert.
-You answer questions about cat health, behavior, diet, and care.
-You also look at photos or videos the user uploads.
+You help cat owners with advice about cat health, food, behavior and care.
+If the user uploads a cat photo or video, respond with relevant helpful tips.
+Be caring, clear and easy to understand.
 """
 
 # ✅ Chat logic
@@ -26,9 +29,10 @@ def chat(user_input, image_file, video_file, history=[]):
     client = Groq(api_key=GROQ_API_KEY)
 
     messages = [{"role": "system", "content": SYSTEM_PROMPT}]
-    for u, b in history:
-        messages.append({"role": "user", "content": u})
-        messages.append({"role": "assistant", "content": b})
+    for user_msg, bot_msg in history:
+        messages.append({"role": "user", "content": user_msg})
+        messages.append({"role": "assistant", "content": bot_msg})
+
     messages.append({"role": "user", "content": user_input + context})
 
     response = client.chat.completions.create(
@@ -37,31 +41,37 @@ def chat(user_input, image_file, video_file, history=[]):
         temperature=0.7,
         max_tokens=512,
     )
+
     reply = response.choices[0].message.content
     history.append((user_input, reply))
     return "", None, None, history
 
-# ✅ UI
-with gr.Blocks() as demo:
+# ✅ Gradio UI
+with gr.Blocks(
+    title="🐱 Dr Cat",
+    theme=gr.themes.Base(primary_hue="pink", secondary_hue="rose")
+) as demo:
     gr.Markdown(
         """
-        <h1>🐱 Dr Cat</h1>
-        <p>Type your question. Optionally upload photo 📸 or video 🎥 of your cat.</p>
+        <div style="text-align:center">
+            <h1>🐱 Dr Cat</h1>
+            <p>Ask me anything about your cat! You can also upload a photo 📸 or video 🎥.</p>
+        </div>
         """
     )
 
-    chatbot = gr.Chatbot(height=400)
-    msg = gr.Textbox(label="Question", placeholder="Type here...")
+    chatbot = gr.Chatbot(label="🐈 Dr Cat", height=400)
+    msg = gr.Textbox(label="Your Question", placeholder="Type your cat question...", lines=2)
     image = gr.Image(label="📸 Upload Cat Photo", type="filepath")
     video = gr.Video(label="🎥 Upload Cat Video", type="filepath")
-    btn = gr.Button("Ask Dr Cat")
+    send_btn = gr.Button("Ask Dr Cat")
 
     state = gr.State([])
 
-    btn.click(
-        chat,
+    send_btn.click(
+        fn=chat,
         inputs=[msg, image, video, state],
-        outputs=[msg, image, video, chatbot]
+        outputs=[msg, image, video, chatbot],
     )
 
 demo.launch()