Update app.py

app.py

@@ -1,30 +1,208 @@
 import os
-import streamlit as st
+import gradio as gr
 from groq import Groq
 
-st.set_page_config(page_title="Sample Chatbot", page_icon="🤖")
 
-st.title("🤖 Jehan Zada Chatboot")
+# ============================================================
+# JEHAN ZADA CHATBOT
+# ============================================================
 
-# Read API key
-api_key = os.getenv("Samplechatboot")
+# Get Groq API key from Hugging Face Secrets
+GROQ_API_KEY = os.environ.get("AIbrainkey")
 
-if not api_key:
-    st.error("API key not found. Please add Samplechatboot in Space secrets.")
-    st.stop()
+# Check if API key exists
+if not GROQ_API_KEY:
+    raise ValueError(
+        "GROQ_API_KEY is missing. "
+        "Please add your Groq API key in Hugging Face Secrets."
+    )
 
-client = Groq(api_key=api_key)
 
-user_input = st.text_input("Ask your question:")
+# Create Groq client
+client = Groq(api_key=GROQ_API_KEY)
 
-if st.button("Send"):
-    if user_input.strip():
-        with st.spinner("Thinking..."):
-            response = client.chat.completions.create(
-                model="llama-3.3-70b-versatile",
-                messages=[{"role": "user", "content": user_input}],
+
+# ============================================================
+# CHATBOT SETTINGS
+# ============================================================
+
+MODEL_NAME = "llama-3.3-70b-versatile"
+
+
+SYSTEM_MESSAGE = """
+You are Jehan Zada Chatbot, a helpful and friendly AI assistant.
+
+Your name is Jehan Zada Chatbot.
+
+Rules:
+- Be helpful, polite and respectful.
+- Give clear and easy-to-understand answers.
+- If the user asks in Urdu, answer in Urdu.
+- If the user asks in English, answer in English.
+- If the user asks in Pashto, answer in Pashto when possible.
+- For difficult topics, explain things step by step.
+- Do not pretend to know something if you are not sure.
+- Keep answers reasonably concise unless the user asks for details.
+"""
+
+
+# ============================================================
+# CHAT FUNCTION
+# ============================================================
+
+def chat_with_jehan(message, history):
+    """
+    Send the user's message to Groq and return the AI response.
+    """
+
+    # If user sends an empty message
+    if not message or not message.strip():
+        return "Please write a message first."
+
+    # Start conversation with system instructions
+    messages = [
+        {
+            "role": "system",
+            "content": SYSTEM_MESSAGE
+        }
+    ]
+
+    # Add previous conversation
+    for user_message, assistant_message in history:
+
+        if user_message:
+            messages.append(
+                {
+                    "role": "user",
+                    "content": user_message
+                }
+            )
+
+        if assistant_message:
+            messages.append(
+                {
+                    "role": "assistant",
+                    "content": assistant_message
+                }
             )
-            st.success("Response:")
-            st.write(response.choices[0].message.content)
-    else:
-        st.warning("Please enter a message.")
+
+    # Add current user message
+    messages.append(
+        {
+            "role": "user",
+            "content": message
+        }
+    )
+
+    try:
+
+        # Send request to Groq
+        response = client.chat.completions.create(
+            model=MODEL_NAME,
+            messages=messages,
+            temperature=0.7,
+            max_tokens=2048
+        )
+
+        # Get AI response
+        answer = response.choices[0].message.content
+
+        return answer
+
+    except Exception as error:
+
+        return (
+            "Sorry, something went wrong.\n\n"
+            f"Error: {str(error)}"
+        )
+
+
+# ============================================================
+# GRADIO INTERFACE
+# ============================================================
+
+with gr.Blocks(
+    title="Jehan Zada Chatbot"
+) as demo:
+
+    gr.Markdown(
+        """
+        # 🤖 Jehan Zada Chatbot
+
+        ### Welcome! I am Jehan Zada Chatbot.
+
+        Ask me anything in **English, Urdu, or Pashto**.
+
+        ---
+        """
+    )
+
+    chatbot = gr.Chatbot(
+        label="Jehan Zada Chatbot",
+        height=500
+    )
+
+    message = gr.Textbox(
+        label="Your Message",
+        placeholder="Type your message here...",
+        lines=2
+    )
+
+    with gr.Row():
+
+        send_button = gr.Button(
+            "Send 🚀",
+            variant="primary"
+        )
+
+        clear_button = gr.Button(
+            "Clear Chat 🗑️"
+        )
+
+
+    # Send message when button is clicked
+    send_button.click(
+        chat_with_jehan,
+        inputs=[message, chatbot],
+        outputs=chatbot
+    ).then(
+        lambda: "",
+        inputs=None,
+        outputs=message
+    )
+
+
+    # Send message when Enter is pressed
+    message.submit(
+        chat_with_jehan,
+        inputs=[message, chatbot],
+        outputs=chatbot
+    ).then(
+        lambda: "",
+        inputs=None,
+        outputs=message
+    )
+
+
+    # Clear conversation
+    clear_button.click(
+        lambda: [],
+        inputs=None,
+        outputs=chatbot
+    )
+
+
+    gr.Markdown(
+        """
+        ---
+        **Jehan Zada Chatbot** | Powered by Groq + Gradio
+        """
+    )
+
+
+# ============================================================
+# START APPLICATION
+# ============================================================
+
+if __name__ == "__main__":
+    demo.launch()