Update app.py

app.py

@@ -7,33 +7,42 @@ from groq import Groq
 # JEHAN ZADA CHATBOT
 # ============================================================
 
-# Your Hugging Face Secret name:
+# Hugging Face Secret name:
 # AIbrainkey
+
 API_KEY = os.environ.get("AIbrainkey")
 
 
-# Check API key
+# ============================================================
+# CHECK API KEY
+# ============================================================
+
 if not API_KEY:
     raise ValueError(
         "AIbrainkey is missing. "
-        "Please add your Groq API key in Hugging Face "
-        "Settings > Secrets."
+        "Please add your Groq API key in "
+        "Hugging Face > Settings > Secrets."
     )
 
 
-# Create Groq client
-client = Groq(api_key=API_KEY)
+# ============================================================
+# GROQ CLIENT
+# ============================================================
+
+client = Groq(
+    api_key=API_KEY
+)
 
 
 # ============================================================
-# GROQ MODEL
+# MODEL
 # ============================================================
 
 MODEL_NAME = "llama-3.3-70b-versatile"
 
 
 # ============================================================
-# CHATBOT PERSONALITY
+# SYSTEM PROMPT
 # ============================================================
 
 SYSTEM_MESSAGE = """
@@ -44,25 +53,26 @@ Your name is Jehan Zada Chatbot.
 You are a helpful, friendly, respectful and intelligent AI
 assistant.
 
-LANGUAGES:
+You can communicate in:
 - 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.
+
+1. If the user writes in English, answer in English.
+2. If the user writes in Urdu, answer in Urdu.
+3. If the user writes in Pashto, answer in Pashto.
+4. If the user mixes languages, respond naturally.
 
 BEHAVIOR:
+
 - Be polite and respectful.
-- Give clear answers.
-- Explain difficult topics in simple language.
+- Give clear and useful answers.
+- Explain difficult subjects in simple language.
 - Help beginners step by step.
-- Do not make up facts.
-- If you do not know something, say so.
+- Do not invent facts.
+- If you are unsure, clearly say that you are unsure.
 - Give examples when useful.
 - Keep normal answers reasonably concise.
 - Give detailed answers when the user asks for details.
@@ -77,11 +87,18 @@ You are an AI chatbot created by Jehan Zada.
 
 def chat_with_jehan(message, history):
 
+    # --------------------------------------------------------
     # Check empty message
+    # --------------------------------------------------------
+
     if not message or not message.strip():
-        return history, ""
+        return "Please write a message first."
+
+
+    # --------------------------------------------------------
+    # Create messages for Groq
+    # --------------------------------------------------------
 
-    # Start with system instructions
     messages = [
         {
             "role": "system",
@@ -89,6 +106,7 @@ def chat_with_jehan(message, history):
         }
     ]
 
+
     # --------------------------------------------------------
     # Add previous conversation
     # --------------------------------------------------------
@@ -97,48 +115,47 @@ def chat_with_jehan(message, history):
 
         for item in history:
 
-            # Old Gradio format:
-            # [user_message, assistant_message]
+            # Dictionary format
+            if isinstance(item, dict):
+
+                role = item.get("role")
+                content = item.get("content")
 
-            if isinstance(item, (list, tuple)):
+                if role in ["user", "assistant"]:
 
-                if len(item) >= 1:
-                    user_message = item[0]
+                    # Simple text
+                    if isinstance(content, str):
 
-                    if user_message:
                         messages.append(
                             {
-                                "role": "user",
-                                "content": str(user_message)
+                                "role": role,
+                                "content": content
                             }
                         )
 
-                if len(item) >= 2:
-                    assistant_message = item[1]
 
-                    if assistant_message:
-                        messages.append(
-                            {
-                                "role": "assistant",
-                                "content": str(assistant_message)
-                            }
-                        )
+            # Old tuple/list format
+            elif isinstance(item, (list, tuple)):
 
-            # New Gradio dictionary format
-            elif isinstance(item, dict):
+                if len(item) >= 1 and item[0]:
 
-                role = item.get("role")
-                content = item.get("content")
+                    messages.append(
+                        {
+                            "role": "user",
+                            "content": str(item[0])
+                        }
+                    )
 
-                if role in ["user", "assistant"] and content:
+                if len(item) >= 2 and item[1]:
 
                     messages.append(
                         {
-                            "role": role,
-                            "content": str(content)
+                            "role": "assistant",
+                            "content": str(item[1])
                         }
                     )
 
+
     # --------------------------------------------------------
     # Add current user message
     # --------------------------------------------------------
@@ -150,195 +167,80 @@ def chat_with_jehan(message, history):
         }
     )
 
+
     # --------------------------------------------------------
-    # Call Groq
+    # Ask Groq
     # --------------------------------------------------------
 
     try:
 
         response = client.chat.completions.create(
+
             model=MODEL_NAME,
+
             messages=messages,
+
             temperature=0.7,
+
             max_tokens=2048
         )
 
+
+        # Get answer
         answer = response.choices[0].message.content
 
+
+        return answer
+
+
     except Exception as error:
 
-        answer = (
-            "Sorry, I could not process your request.\n\n"
-            "Please check your Groq API key and try again.\n\n"
+        return (
+            "❌ Sorry, I could not process your request.\n\n"
+            "Please check your Groq API key.\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
+# GRADIO CHAT INTERFACE
 # ============================================================
 
-def clear_chat():
-    return [], ""
-
+demo = gr.ChatInterface(
 
-# ============================================================
-# GRADIO USER INTERFACE
-# ============================================================
+    fn=chat_with_jehan,
 
-with gr.Blocks(
-    title="Jehan Zada Chatbot"
-) as demo:
+    title="🤖 Jehan Zada Chatbot",
 
-    # --------------------------------------------------------
-    # Header
-    # --------------------------------------------------------
+    description="""
+Welcome to **Jehan Zada Chatbot**!
 
-    gr.Markdown(
-        """
-# 🤖 Jehan Zada Chatbot
-
-### Welcome!
-
-I am **Jehan Zada Chatbot**.
-
-You can ask me questions in:
+You can ask 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
-    )
+    examples=[
+        "Hello, who are you?",
+        "پاکستان کیا ہے؟",
+        "مصنوعی ذہانت کیا ہے؟",
+        "ما ته په پښتو کې ځواب راکړه",
+        "Explain Python in simple words."
+    ],
 
-    # --------------------------------------------------------
-    # Buttons
-    # --------------------------------------------------------
+    submit_btn="Send 🚀",
 
-    with gr.Row():
-
-        send_button = gr.Button(
-            "Send 🚀",
-            variant="primary"
-        )
-
-        clear_button = gr.Button(
-            "Clear Chat 🗑️"
-        )
-
-    # --------------------------------------------------------
-    # Examples
-    # --------------------------------------------------------
-
-    gr.Examples(
-        examples=[
-            "Hello, who are you?",
-            "پاکستان کیا ہے؟",
-            "مصنوعی ذہانت کیا ہے؟",
-            "ما ته په پښتو کې ځواب راکړه",
-            "Explain Python in simple words."
-        ],
-        inputs=message
-    )
-
-    # --------------------------------------------------------
-    # Send button
-    # --------------------------------------------------------
-
-    send_button.click(
-        fn=chat_with_jehan,
-        inputs=[
-            message,
-            chatbot
-        ],
-        outputs=[
-            chatbot,
-            message
-        ]
-    )
-
-    # --------------------------------------------------------
-    # Enter key
-    # --------------------------------------------------------
-
-    message.submit(
-        fn=chat_with_jehan,
-        inputs=[
-            message,
-            chatbot
-        ],
-        outputs=[
-            chatbot,
-            message
-        ]
-    )
-
-    # --------------------------------------------------------
-    # Clear button
-    # --------------------------------------------------------
-
-    clear_button.click(
-        fn=clear_chat,
-        inputs=None,
-        outputs=[
-            chatbot,
-            message
-        ]
-    )
-
-    # --------------------------------------------------------
-    # Footer
-    # --------------------------------------------------------
-
-    gr.Markdown(
-        """
----
-
-### Jehan Zada Chatbot
-
-Powered by **Groq + Gradio**
-
-"""
-    )
+    clear_btn="Clear Chat 🗑️"
+)
 
 
 # ============================================================
-# START APP
+# START
 # ============================================================
 
 if __name__ == "__main__":
+
     demo.launch()