Spaces:
Running
Running
| import os | |
| import gradio as gr | |
| from groq import Groq | |
| # ============================================================ | |
| # JEHAN ZADA CHATBOT | |
| # ============================================================ | |
| # Your Hugging Face Secret name is: | |
| # AIbrainkey | |
| API_KEY = os.environ.get("AIbrainkey") | |
| # ============================================================ | |
| # CHECK API KEY | |
| # ============================================================ | |
| if not API_KEY: | |
| raise ValueError( | |
| "AIbrainkey is missing.\n\n" | |
| "Please go to Hugging Face:\n" | |
| "Settings > Variables and secrets > Secrets\n\n" | |
| "and add your Groq API key with the name:\n" | |
| "AIbrainkey" | |
| ) | |
| # ============================================================ | |
| # GROQ CLIENT | |
| # ============================================================ | |
| client = Groq( | |
| api_key=API_KEY | |
| ) | |
| # ============================================================ | |
| # CURRENT GROQ MODEL | |
| # ============================================================ | |
| MODEL_NAME = "openai/gpt-oss-120b" | |
| # ============================================================ | |
| # JEHAN ZADA CHATBOT PERSONALITY | |
| # ============================================================ | |
| SYSTEM_MESSAGE = """ | |
| You are Jehan Zada Chatbot. | |
| Your name is Jehan Zada Chatbot. | |
| You are a helpful, friendly, respectful and intelligent AI | |
| assistant. | |
| You can communicate in: | |
| 1. English | |
| 2. Urdu | |
| 3. 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 using the | |
| language that best matches the user's question. | |
| BEHAVIOR: | |
| - Be polite and respectful. | |
| - Give clear and useful answers. | |
| - Explain difficult subjects in simple language. | |
| - Help beginners step by step. | |
| - Do not invent facts. | |
| - If you are unsure about something, say so. | |
| - Give examples when useful. | |
| - Keep normal answers reasonably concise. | |
| - Give detailed answers when the user asks for details. | |
| - Use headings and bullet points when helpful. | |
| IDENTITY: | |
| You are Jehan Zada Chatbot, | |
| an AI assistant created by Jehan Zada. | |
| """ | |
| # ============================================================ | |
| # CHAT FUNCTION | |
| # ============================================================ | |
| def chat_with_jehan(message, history): | |
| # -------------------------------------------------------- | |
| # Check empty message | |
| # -------------------------------------------------------- | |
| if not message or not message.strip(): | |
| return "Please write a message first." | |
| # -------------------------------------------------------- | |
| # Create conversation | |
| # -------------------------------------------------------- | |
| messages = [ | |
| { | |
| "role": "system", | |
| "content": SYSTEM_MESSAGE | |
| } | |
| ] | |
| # -------------------------------------------------------- | |
| # Add previous conversation | |
| # -------------------------------------------------------- | |
| if history: | |
| for item in history: | |
| # ----------------------------------------------- | |
| # Dictionary format | |
| # ----------------------------------------------- | |
| if isinstance(item, dict): | |
| role = item.get("role") | |
| content = item.get("content") | |
| if role in ["user", "assistant"] and content: | |
| if isinstance(content, str): | |
| messages.append( | |
| { | |
| "role": role, | |
| "content": content | |
| } | |
| ) | |
| # ----------------------------------------------- | |
| # Old list/tuple format | |
| # ----------------------------------------------- | |
| elif isinstance(item, (list, tuple)): | |
| # User message | |
| if len(item) >= 1 and item[0]: | |
| messages.append( | |
| { | |
| "role": "user", | |
| "content": str(item[0]) | |
| } | |
| ) | |
| # Assistant message | |
| if len(item) >= 2 and item[1]: | |
| messages.append( | |
| { | |
| "role": "assistant", | |
| "content": str(item[1]) | |
| } | |
| ) | |
| # -------------------------------------------------------- | |
| # Add current user message | |
| # -------------------------------------------------------- | |
| messages.append( | |
| { | |
| "role": "user", | |
| "content": message | |
| } | |
| ) | |
| # ======================================================== | |
| # SEND REQUEST TO GROQ | |
| # ======================================================== | |
| try: | |
| response = client.chat.completions.create( | |
| model=MODEL_NAME, | |
| messages=messages, | |
| temperature=0.7, | |
| max_tokens=2048 | |
| ) | |
| # ---------------------------------------------------- | |
| # Get AI answer | |
| # ---------------------------------------------------- | |
| answer = response.choices[0].message.content | |
| # ---------------------------------------------------- | |
| # Return answer | |
| # ---------------------------------------------------- | |
| return answer | |
| # ======================================================== | |
| # ERROR HANDLING | |
| # ======================================================== | |
| except Exception as error: | |
| return ( | |
| "❌ Sorry, I could not process your request.\n\n" | |
| "Please try again.\n\n" | |
| "Technical error:\n" | |
| + str(error) | |
| ) | |
| # ============================================================ | |
| # GRADIO CHAT INTERFACE | |
| # ============================================================ | |
| demo = gr.ChatInterface( | |
| fn=chat_with_jehan, | |
| title="🤖 Jehan Zada Chatbot", | |
| description=""" | |
| Welcome to **Jehan Zada Chatbot**! | |
| You can ask questions in: | |
| 🇬🇧 English | |
| 🇵🇰 Urdu | |
| 🟢 Pashto | |
| """, | |
| examples=[ | |
| "Hello, who are you?", | |
| "پاکستان کیا ہے؟", | |
| "مصنوعی ذہانت کیا ہے؟", | |
| "ما ته په پښتو کې ځواب راکړه", | |
| "Explain Python in simple words." | |
| ] | |
| ) | |
| # ============================================================ | |
| # START APPLICATION | |
| # ============================================================ | |
| if __name__ == "__main__": | |
| demo.launch() |