Update app.py

app.py

@@ -1,82 +1,85 @@
 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"
 
 SYSTEM_HI = "तुम एक सहायक हिन्दी सहायक हो। स्पष्ट, संक्षिप्त और सही उत्तर दो।"
 SYSTEM_BN = "তুমি একজন সহায়ক বাংলা সহকারী। সংক্ষিপ্ত, স্পষ্ট এবং সঠিক উত্তর দাও।"
 
-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
-
-    # Check API key
+def call_openai(messages):
     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)]
+        raise RuntimeError("OPENAI_API_KEY is not set in Settings → Variables and secrets")
 
-    # 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}")]
+    # Lazy import so startup doesn't crash if deps missing
+    from openai import OpenAI
+    client = OpenAI(api_key=api_key)
+
+    resp = client.chat.completions.create(
+        model=MODEL,
+        messages=messages,
+        temperature=0.6,
+        max_tokens=400,
+    )
+    return resp.choices[0].message.content
+
+def build_messages(history, lang_label):
+    # lang_label is "हिन्दी" or "বাংলা"
+    lang = "bn" if lang_label == "বাংলা" else "hi"
+    sys = SYSTEM_BN if lang == "bn" else SYSTEM_HI
 
-    # Convert history to OpenAI messages
     messages = [{"role": "system", "content": sys}]
     for user, bot in history:
         if user:
-            messages.append({"role": "user", content_or_key("content"): user})
+            messages.append({"role": "user", "content": user})
         if bot:
             messages.append({"role": "assistant", "content": bot})
+    return messages
 
+def on_user_submit(user_text, chat_state):
+    """Append the user message; bot will fill the reply in the next step."""
+    if not user_text.strip():
+        return "", chat_state
+    chat_state = chat_state + [(user_text, None)]
+    return "", chat_state
+
+def on_bot_reply(chat_state, lang_label):
+    """Call the model and set the reply for the last turn."""
     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)]
+        messages = build_messages(chat_state, lang_label)
+        answer = call_openai(messages)
+        # put the bot answer into the last tuple
+        last_user, _ = chat_state[-1]
+        chat_state[-1] = (last_user, answer)
+        return chat_state
     except Exception as e:
-        err = f"API error: {e}"
-        return history + [("", err if lang == "hi" else f"API ত্রুটি: {e}")]
-
-def content_or_key(key_name):
-    # tiny helper to satisfy static analyzers; keeps code readable above
-    return key_name
+        last_user, _ = chat_state[-1]
+        err_hi = f"त्रुटि: {e}"
+        err_bn = f"ত্রুটি: {e}"
+        chat_state[-1] = (last_user, err_bn if lang_label == "বাংলা" else err_hi)
+        return chat_state
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("### 🇮🇳 Indic Chat — हिंदी / বাংলা")
+
     lang = gr.Radio(
         choices=["हिन्दी", "বাংলা"],
         value="हिन्दी",
-        label="Language"
+        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 on_user_submit(user_text, chat_state):
-        if not user_text.strip():
-            return "", chat_state
-        return "", chat_state + [(user_text, None)]
+    # ✅ Use 'tuples' (plural)
+    chat = gr.Chatbot(height=420, type="tuples")
 
-    def on_bot_reply(chat_state, lang_label):
-        return chat_fn(chat_state, lang_label)
+    txt = gr.Textbox(placeholder="Type in Hindi or Bangla…", autofocus=True)
+    send = gr.Button("Send")
 
-    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])
+    # Wire up: user adds a turn → then bot replies
+    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()