app.py

app.py

@@ -1,43 +1,82 @@
-import os, gradio as gr
-from openai import OpenAI
+import os
+import gradio as gr
+
+# We import lazily inside the function to avoid crashing on startup if the key isn't set.
+MODEL = "gpt-4o-mini"
 
-MODEL = "gpt-4o-mini"  # multilingual, great for Hindi/Bangla
 SYSTEM_HI = "तुम एक सहायक हिन्दी सहायक हो। स्पष्ट, संक्षिप्त और सही उत्तर दो।"
 SYSTEM_BN = "তুমি একজন সহায়ক বাংলা সহকারী। সংক্ষিপ্ত, স্পষ্ট এবং সঠিক উত্তর দাও।"
 
-client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
-
-def chat_fn(history, lang):
+def chat_fn(history, lang_label):
+    """
+    history: list of (user, bot) tuples from gr.Chatbot
+    lang_label: "हिन्दी" or "বাংলা"
+    """
+    # Map UI label -> code
+    lang = "bn" if lang_label == "বাংলা" else "hi"
     sys = SYSTEM_BN if lang == "bn" else SYSTEM_HI
-    # convert Gradio history -> OpenAI messages
-    msgs = [{"role": "system", "content": sys}]
+
+    # Check API key
+    api_key = os.getenv("OPENAI_API_KEY", "").strip()
+    if not api_key:
+        msg = ("❗ OPENAI_API_KEY not set in this Space’s Settings → Variables and secrets."
+               if lang == "hi" else
+               "❗ এই স্পেসের Settings → Variables and secrets এ OPENAI_API_KEY সেট করা হয়নি।")
+        return history + [("", msg)]
+
+    # Import + client only when needed (safer on cold start)
+    try:
+        from openai import OpenAI
+        client = OpenAI(api_key=api_key)
+    except Exception as e:
+        err = f"OpenAI client error: {e}"
+        return history + [("", err if lang == "hi" else f"OpenAI ক্লায়েন্ট ত্রুটি: {e}")]
+
+    # Convert history to OpenAI messages
+    messages = [{"role": "system", "content": sys}]
     for user, bot in history:
-        msgs.append({"role": "user", "content": user})
+        if user:
+            messages.append({"role": "user", content_or_key("content"): user})
         if bot:
-            msgs.append({"role": "assistant", "content": bot})
+            messages.append({"role": "assistant", "content": bot})
+
+    try:
+        resp = client.chat.completions.create(
+            model=MODEL,
+            messages=messages,
+            temperature=0.6,
+            max_tokens=400,
+        )
+        answer = resp.choices[0].message.content
+        return history + [("", answer)]
+    except Exception as e:
+        err = f"API error: {e}"
+        return history + [("", err if lang == "hi" else f"API ত্রুটি: {e}")]
 
-    resp = client.chat.completions.create(model=MODEL, messages=msgs, temperature=0.6, max_tokens=400)
-    return resp.choices[0].message.content
+def content_or_key(key_name):
+    # tiny helper to satisfy static analyzers; keeps code readable above
+    return key_name
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("### 🇮🇳 Indic Chat — हिंदी / বাংলা")
-    with gr.Row():
-        lang = gr.Radio(choices=[("हिन्दी","hi"), ("বাংলা","bn")], value=("हिन्दी","hi"), label="Language")
+    lang = gr.Radio(
+        choices=["हिन्दी", "বাংলা"],
+        value="हिन्दी",
+        label="Language"
+    )
     chat = gr.Chatbot(height=420, type="tuple")
     txt = gr.Textbox(placeholder="Type in Hindi or Bangla…", autofocus=True)
     send = gr.Button("Send")
 
-    def user_send(user_text, chat_state):
-        if not user_text.strip(): return gr.update(), chat_state
-        chat_state = chat_state + [(user_text, None)]
-        return "", chat_state
+    def on_user_submit(user_text, chat_state):
+        if not user_text.strip():
+            return "", chat_state
+        return "", chat_state + [(user_text, None)]
 
-    def bot_reply(chat_state, lang_val):
-        answer = chat_fn(chat_state, lang_val[1])
-        chat_state[-1] = (chat_state[-1][0], answer)
-        return chat_state
+    def on_bot_reply(chat_state, lang_label):
+        return chat_fn(chat_state, lang_label)
 
-    txt.submit(user_send, [txt, chat], [txt, chat]).then(bot_reply, [chat, lang], [chat])
-    send.click(user_send, [txt, chat], [txt, chat]).then(bot_reply, [chat, lang], [chat])
+    txt.submit(on_user_submit, [txt, chat], [txt, chat]).then(on_bot_reply, [chat, lang], [chat])
+    send.click(on_user_submit, [txt, chat], [txt, chat]).then(on_bot_reply, [chat, lang], [chat])
 
 demo.launch()