Update app.py

app.py

@@ -7,15 +7,17 @@ from groq import Groq
 # JEHAN ZADA CHATBOT
 # ============================================================
 
-# Your Hugging Face Secret name
+# Your Hugging Face Secret name:
+# AIbrainkey
 API_KEY = os.environ.get("AIbrainkey")
 
 
 # Check API key
 if not API_KEY:
     raise ValueError(
-        "AIbrainkey not found. "
-        "Please add your Groq API key in Hugging Face Secrets."
+        "AIbrainkey is missing. "
+        "Please add your Groq API key in Hugging Face "
+        "Settings > Secrets."
     )
 
 
@@ -24,14 +26,14 @@ client = Groq(api_key=API_KEY)
 
 
 # ============================================================
-# MODEL
+# GROQ MODEL
 # ============================================================
 
 MODEL_NAME = "llama-3.3-70b-versatile"
 
 
 # ============================================================
-# SYSTEM PROMPT
+# CHATBOT PERSONALITY
 # ============================================================
 
 SYSTEM_MESSAGE = """
@@ -39,26 +41,33 @@ You are Jehan Zada Chatbot.
 
 Your name is Jehan Zada Chatbot.
 
-You are a helpful, friendly and respectful AI assistant.
-
-You can communicate in:
-English
-Urdu
-Pashto
-
-Language rules:
-- 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.
-
-Always:
-- Be helpful and polite.
-- Explain difficult things in simple language.
+You are a helpful, friendly, respectful and intelligent AI
+assistant.
+
+LANGUAGES:
+- English
+- Urdu
+- Pashto
+
+LANGUAGE RULES:
+1. If the user asks in English, answer in English.
+2. If the user asks in Urdu, answer in Urdu.
+3. If the user asks in Pashto, answer in Pashto.
+4. If the user uses a mixture of languages, respond naturally
+   using the language that best matches the user's message.
+
+BEHAVIOR:
+- Be polite and respectful.
+- Give clear answers.
+- Explain difficult topics in simple language.
 - Help beginners step by step.
-- Do not make up information.
-- If you are not sure, say so.
+- Do not make up facts.
+- If you do not know something, say so.
+- Give examples when useful.
+- Keep normal answers reasonably concise.
 - Give detailed answers when the user asks for details.
+
+You are an AI chatbot created by Jehan Zada.
 """
 
 
@@ -68,11 +77,11 @@ Always:
 
 def chat_with_jehan(message, history):
 
-    # Empty message
+    # Check empty message
     if not message or not message.strip():
-        return "Please write a message first."
+        return history, ""
 
-    # System message
+    # Start with system instructions
     messages = [
         {
             "role": "system",
@@ -80,7 +89,10 @@ def chat_with_jehan(message, history):
         }
     ]
 
+    # --------------------------------------------------------
     # Add previous conversation
+    # --------------------------------------------------------
+
     if history:
 
         for item in history:
@@ -90,30 +102,36 @@ def chat_with_jehan(message, history):
 
             if isinstance(item, (list, tuple)):
 
-                if len(item) >= 1 and item[0]:
-                    messages.append(
-                        {
-                            "role": "user",
-                            "content": str(item[0])
-                        }
-                    )
-
-                if len(item) >= 2 and item[1]:
-                    messages.append(
-                        {
-                            "role": "assistant",
-                            "content": str(item[1])
-                        }
-                    )
-
-            # Newer Gradio format:
-            # {"role": "...", "content": "..."}
+                if len(item) >= 1:
+                    user_message = item[0]
+
+                    if user_message:
+                        messages.append(
+                            {
+                                "role": "user",
+                                "content": str(user_message)
+                            }
+                        )
+
+                if len(item) >= 2:
+                    assistant_message = item[1]
+
+                    if assistant_message:
+                        messages.append(
+                            {
+                                "role": "assistant",
+                                "content": str(assistant_message)
+                            }
+                        )
+
+            # New Gradio dictionary format
             elif isinstance(item, dict):
 
                 role = item.get("role")
                 content = item.get("content")
 
                 if role in ["user", "assistant"] and content:
+
                     messages.append(
                         {
                             "role": role,
@@ -121,7 +139,10 @@ def chat_with_jehan(message, history):
                         }
                     )
 
-    # Add current message
+    # --------------------------------------------------------
+    # Add current user message
+    # --------------------------------------------------------
+
     messages.append(
         {
             "role": "user",
@@ -129,9 +150,12 @@ def chat_with_jehan(message, history):
         }
     )
 
+    # --------------------------------------------------------
+    # Call Groq
+    # --------------------------------------------------------
+
     try:
 
-        # Call Groq
         response = client.chat.completions.create(
             model=MODEL_NAME,
             messages=messages,
@@ -139,51 +163,92 @@ def chat_with_jehan(message, history):
             max_tokens=2048
         )
 
-        # Get answer
         answer = response.choices[0].message.content
 
-        return answer
-
     except Exception as error:
 
-        return (
-            "Sorry, I could not answer your question.\n\n"
-            "Please check your Groq API key or try again.\n\n"
-            "Error: " + str(error)
+        answer = (
+            "Sorry, I could not process your request.\n\n"
+            "Please check your Groq API key and try again.\n\n"
+            "Technical error:\n"
+            + str(error)
         )
 
+    # --------------------------------------------------------
+    # Add conversation to chat history
+    # --------------------------------------------------------
+
+    new_history = list(history) if history else []
+
+    new_history.append(
+        [message, answer]
+    )
+
+    # Return updated chat + empty textbox
+    return new_history, ""
+
+
+# ============================================================
+# CLEAR CHAT
+# ============================================================
+
+def clear_chat():
+    return [], ""
+
 
 # ============================================================
-# USER INTERFACE
+# GRADIO USER INTERFACE
 # ============================================================
 
 with gr.Blocks(
     title="Jehan Zada Chatbot"
 ) as demo:
 
+    # --------------------------------------------------------
+    # Header
+    # --------------------------------------------------------
+
     gr.Markdown(
         """
-        # 🤖 Jehan Zada Chatbot
+# 🤖 Jehan Zada Chatbot
 
-        ### Welcome!
+### Welcome!
 
-        I can answer your questions in **English, Urdu and Pashto**.
+I am **Jehan Zada Chatbot**.
 
-        ---
-        """
+You can ask me questions in:
+
+🇬🇧 English  
+🇵🇰 Urdu  
+🟢 Pashto
+
+---
+"""
     )
 
+    # --------------------------------------------------------
+    # Chat window
+    # --------------------------------------------------------
+
     chatbot = gr.Chatbot(
         label="Jehan Zada Chatbot",
         height=500
     )
 
+    # --------------------------------------------------------
+    # Message box
+    # --------------------------------------------------------
+
     message = gr.Textbox(
         label="Your Message",
         placeholder="Type your message here...",
         lines=2
     )
 
+    # --------------------------------------------------------
+    # Buttons
+    # --------------------------------------------------------
+
     with gr.Row():
 
         send_button = gr.Button(
@@ -195,56 +260,85 @@ with gr.Blocks(
             "Clear Chat 🗑️"
         )
 
+    # --------------------------------------------------------
+    # Examples
+    # --------------------------------------------------------
+
+    gr.Examples(
+        examples=[
+            "Hello, who are you?",
+            "پاکستان کیا ہے؟",
+            "مصنوعی ذہانت کیا ہے؟",
+            "ما ته په پښتو کې ځواب راکړه",
+            "Explain Python in simple words."
+        ],
+        inputs=message
+    )
 
+    # --------------------------------------------------------
     # Send button
+    # --------------------------------------------------------
+
     send_button.click(
-        chat_with_jehan,
-        inputs=[message, chatbot],
-        outputs=[chatbot]
-    ).then(
-        lambda: "",
-        inputs=None,
-        outputs=[message]
+        fn=chat_with_jehan,
+        inputs=[
+            message,
+            chatbot
+        ],
+        outputs=[
+            chatbot,
+            message
+        ]
     )
 
+    # --------------------------------------------------------
+    # Enter key
+    # --------------------------------------------------------
 
-    # Press Enter
     message.submit(
-        chat_with_jehan,
-        inputs=[message, chatbot],
-        outputs=[chatbot]
-    ).then(
-        lambda: "",
-        inputs=None,
-        outputs=[message]
+        fn=chat_with_jehan,
+        inputs=[
+            message,
+            chatbot
+        ],
+        outputs=[
+            chatbot,
+            message
+        ]
     )
 
+    # --------------------------------------------------------
+    # Clear button
+    # --------------------------------------------------------
 
-    # Clear chat
     clear_button.click(
-        lambda: [],
+        fn=clear_chat,
         inputs=None,
-        outputs=[chatbot]
+        outputs=[
+            chatbot,
+            message
+        ]
     )
 
+    # --------------------------------------------------------
+    # Footer
+    # --------------------------------------------------------
 
     gr.Markdown(
         """
-        ---
+---
 
-        **Jehan Zada Chatbot**  
-        Powered by Groq + Gradio
-        """
+### Jehan Zada Chatbot
+
+Powered by **Groq + Gradio**
+
+"""
     )
 
 
 # ============================================================
-# START
+# START APP
 # ============================================================
 
 if __name__ == "__main__":
-
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860
-    )
+    demo.launch()