Create app.py

app.py

@@ -1,85 +1,39 @@
-# app.py
 import os
 import gradio as gr
 from google import genai
 
-# Read API key from env (set this as a Space secret on HF: GEMINI_API_KEY)
-api_key = os.environ.get("GEMINI_API_KEY")
-client = genai.Client(api_key=api_key)
-MODEL = os.environ.get("GEMINI_MODEL", "gemini-2.5-flash")
+# ---- Load API key safely ----
+API_KEY = os.getenv("GEMINI_API_KEY")
+if not API_KEY:
+    raise ValueError("Environment variable GEMINI_API_KEY not found or empty")
 
+# ---- Initialize Gemini client ----
+client = genai.Client(api_key=API_KEY)
+MODEL = os.getenv("GEMINI_MODEL", "gemini-2.5-flash")
 
-def call_gemini(prompt: str) -> str:
-    """Call Gemini synchronously and return text reply (safe fallback)."""
+
+def chat_fn(message, history):
+    if not message.strip():
+        return history, history
     try:
-        response = client.models.generate_content(
-            model=MODEL,
-            contents=prompt,
-        )
-        # SDK commonly exposes textual output as .text
-        return getattr(response, "text", str(response))
+        result = client.models.generate_content(model=MODEL, contents=message)
+        reply = getattr(result, "text", str(result))
     except Exception as e:
-        return f"[Error calling Gemini API: {e}]"
-
-
-def generate_reply(message: str, history: list) -> list:
-    """
-    Given a user message and current `history` (list of role/content dicts),
-    append the user's message, call Gemini, append the assistant reply, and
-    return the updated history (in Gradio 'messages' format).
-    """
-    if history is None:
-        history = []
-
-    user_message = message.strip()
-    if not user_message:
-        return history
-
-    # Append user message (messages format)
-    history.append({"role": "user", "content": user_message})
-
-    # Option A: send only the current user message as prompt:
-    # reply_text = call_gemini(user_message)
-
-    # Option B: (commented) send the full conversation as a single prompt for more context.
-    # Uncomment if you want multi-turn context included.
-    # combined_prompt = "\n".join(
-    #     f"{m['role']}: {m['content']}" for m in history
-    # )
-    # reply_text = call_gemini(combined_prompt)
-
-    # Using Option A by default:
-    reply_text = call_gemini(user_message)
-
-    # Append assistant response
-    history.append({"role": "assistant", "content": reply_text})
-    return history
+        reply = f"[Error: {e}]"
+    history = (history or []) + [
+        {"role": "user", "content": message},
+        {"role": "assistant", "content": reply},
+    ]
+    return history, history
 
 
 with gr.Blocks(title="Gemini Chatbot") as demo:
-    gr.Markdown("# Gemini Chatbot (Gradio — messages format)")
-
-    # Use the new 'messages' type so Gradio won't warn about tuples->messages deprecation
-    chatbot = gr.Chatbot(label="Gemini", type="messages")
-    state = gr.State([])  # will hold the list of {"role","content"} dicts
-
-    with gr.Row():
-        txt = gr.Textbox(
-            show_label=False,
-            placeholder="Type your message and press Enter...",
-            lines=1,
-        )
-
-    def user_submit(message, history):
-        # Ensure history is a list
-        if history is None:
-            history = []
-        updated = generate_reply(message, history)
-        # Gradio expects (chatbot, state) outputs; we return the updated messages list for both
-        return updated, updated
-
-    txt.submit(fn=user_submit, inputs=[txt, state], outputs=[chatbot, state])
-    gr.Button("Clear").click(lambda: ([], []), None, [chatbot, state], queue=False)
+    gr.Markdown("## Gemini Chatbot (Gradio + Hugging Face Space)")
+    chatbot = gr.Chatbot(type="messages", label="Gemini")
+    state = gr.State([])
+    txt = gr.Textbox(placeholder="Ask something...", show_label=False)
+    txt.submit(chat_fn, [txt, state], [chatbot, state])
+    gr.Button("Clear").click(lambda: ([], []), None, [chatbot, state])
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT", 7860)))
+    demo.launch(server_name="0.0.0.0", server_port=int(os.getenv("PORT", 7860)))