Create app.py

app.py

@@ -0,0 +1,85 @@
+# 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")
+
+
+def call_gemini(prompt: str) -> str:
+    """Call Gemini synchronously and return text reply (safe fallback)."""
+    try:
+        response = client.models.generate_content(
+            model=MODEL,
+            contents=prompt,
+        )
+        # SDK commonly exposes textual output as .text
+        return getattr(response, "text", str(response))
+    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
+
+
+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)
+
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT", 7860)))