Update app.py

app.py

@@ -7,42 +7,63 @@ from groq import Groq
 # JEHAN ZADA CHATBOT
 # ============================================================
 
-# Get Groq API key from Hugging Face Secrets
-GROQ_API_KEY = os.environ.get("AIbrainkey")
+# Get Groq API key from Hugging Face Secret
+# IMPORTANT:
+# Your Hugging Face Secret name must be: AIbrainkey
+API_KEY = os.environ.get("AIbrainkey")
 
-# Check if API key exists
-if not GROQ_API_KEY:
+
+# Check API key
+if not API_KEY:
     raise ValueError(
-        "GROQ_API_KEY is missing. "
-        "Please add your Groq API key in Hugging Face Secrets."
+        "AIbrainkey was not found. "
+        "Please add your Groq API key in "
+        "Hugging Face > Settings > Secrets."
     )
 
 
 # Create Groq client
-client = Groq(api_key=GROQ_API_KEY)
+client = Groq(api_key=API_KEY)
 
 
 # ============================================================
-# CHATBOT SETTINGS
+# AI MODEL
 # ============================================================
 
 MODEL_NAME = "llama-3.3-70b-versatile"
 
 
+# ============================================================
+# JEHAN ZADA CHATBOT PERSONALITY
+# ============================================================
+
 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.
+You can communicate in:
+- English
+- Urdu
+- Pashto
+
+Language rule:
+- If the user writes in English, answer in English.
+- If the user writes in Urdu, answer in Urdu.
+- If the user writes in Pashto, answer in Pashto.
+- If the user mixes languages, respond naturally in the language that is most appropriate.
+
+Your behavior:
+- Be helpful, polite, respectful and friendly.
+- Explain difficult things in simple language.
+- When a beginner asks a technical question, explain it step by step.
+- Do not make up facts.
+- If you are uncertain, clearly say that you are uncertain.
+- Keep normal answers reasonably concise.
+- Give detailed answers when the user asks for details.
+- Use headings and bullet points when they make the answer easier to understand.
+
+You are an AI chatbot created by Jehan Zada.
 """
 
 
@@ -52,14 +73,15 @@ Rules:
 
 def chat_with_jehan(message, history):
     """
-    Send the user's message to Groq and return the AI response.
+    Send the user's message and previous conversation
+    to Groq and return the AI response.
     """
 
-    # If user sends an empty message
+    # Check empty message
     if not message or not message.strip():
         return "Please write a message first."
 
-    # Start conversation with system instructions
+    # Start messages with system instructions
     messages = [
         {
             "role": "system",
@@ -68,25 +90,26 @@ def chat_with_jehan(message, history):
     ]
 
     # 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
-                }
-            )
-
-    # Add current user message
+    if history:
+        for item in history:
+
+            # Get role
+            role = item.get("role")
+
+            # Get content
+            content = item.get("content")
+
+            # We only need normal text messages
+            if role in ["user", "assistant"] and isinstance(content, str):
+
+                messages.append(
+                    {
+                        "role": role,
+                        "content": content
+                    }
+                )
+
+    # Add the new user message
     messages.append(
         {
             "role": "user",
@@ -96,7 +119,7 @@ def chat_with_jehan(message, history):
 
     try:
 
-        # Send request to Groq
+        # Send conversation to Groq
         response = client.chat.completions.create(
             model=MODEL_NAME,
             messages=messages,
@@ -104,100 +127,61 @@ def chat_with_jehan(message, history):
             max_tokens=2048
         )
 
-        # Get AI response
+        # Get AI answer
         answer = response.choices[0].message.content
 
         return answer
 
     except Exception as error:
 
+        # Show a simple error message
         return (
-            "Sorry, something went wrong.\n\n"
-            f"Error: {str(error)}"
+            "Sorry, I could not process your request.\n\n"
+            "Please check your Groq API key and try again.\n\n"
+            f"Technical error: {str(error)}"
         )
 
 
 # ============================================================
-# GRADIO INTERFACE
+# GRADIO CHATBOT
 # ============================================================
 
-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
-    )
+demo = gr.ChatInterface(
+    fn=chat_with_jehan,
 
-    with gr.Row():
+    title="🤖 Jehan Zada Chatbot",
 
-        send_button = gr.Button(
-            "Send 🚀",
-            variant="primary"
-        )
+    description=(
+        "Welcome to Jehan Zada Chatbot! "
+        "You can ask questions in English, Urdu, or Pashto."
+    ),
 
-        clear_button = gr.Button(
-            "Clear Chat 🗑️"
+    chatbot=gr.Chatbot(
+        type="messages",
+        height=550,
+        placeholder=(
+            "<h2>🤖 Jehan Zada Chatbot</h2>"
+            "<p>Ask me anything...</p>"
         )
+    ),
 
-
-    # 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
-        """
-    )
+    textbox=gr.Textbox(
+        placeholder="Type your message here...",
+        container=True,
+        scale=7
+    ),
+
+    submit_btn="Send 🚀",
+    stop_btn="Stop",
+    clear_btn="Clear Chat 🗑️",
+
+    examples=[
+        "Hello, who are you?",
+        "پاکستان کیا ہے؟",
+        "ما ته په پښتو کې ځواب راکړه",
+        "Explain artificial intelligence in simple words."
+    ]
+)
 
 
 # ============================================================
@@ -205,4 +189,8 @@ with gr.Blocks(
 # ============================================================
 
 if __name__ == "__main__":
-    demo.launch()
+
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860
+    )