Update app.py

app.py

@@ -1,39 +1,40 @@
 import os
 import gradio as gr
-from google import genai
+from google import generativeai as genai
 
-# --- Load Gemini API key from Hugging Face secret ---
+# --- Load Gemini API key securely ---
 GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
 if not GEMINI_API_KEY:
-    raise ValueError("❌ GEMINI_API_KEY not found in environment. Add it in Hugging Face Secrets.")
+    raise ValueError("❌ GEMINI_API_KEY not found. Set it in Hugging Face Secrets.")
 
-# --- Configure Gemini client ---
+# --- Configure GenAI ---
 genai.configure(api_key=GEMINI_API_KEY)
-client = genai.Client()
 
-# Create a persistent chat session
-chat = client.chats.create(model="gemini-2.5-flash")
+# --- Initialize Gemini model ---
+model = genai.GenerativeModel("gemini-2.5-flash")
+chat_session = model.start_chat()
 
-# --- Define chatbot function ---
+# --- Define chat function ---
 def chat_with_gemini(user_input, history):
     try:
-        response = chat.send_message(user_input)
+        response = chat_session.send_message(user_input)
         history.append((user_input, response.text))
         return history, history
     except Exception as e:
-        return history + [(user_input, f"⚠️ Error: {str(e)}")], history
+        history.append((user_input, f"⚠️ Error: {str(e)}"))
+        return history, history
 
 # --- Gradio UI ---
 with gr.Blocks(title="Gemini 2.5 Flash Chatbot") as demo:
-    gr.Markdown("# 🤖 Gemini 2.5 Flash Chatbot\nAsk me anything!")
+    gr.Markdown("## 🤖 Gemini 2.5 Flash Chatbot\nPowered by Google Generative AI")
     chatbot = gr.Chatbot()
-    message = gr.Textbox(placeholder="Type your message and press Enter")
-    clear = gr.Button("Clear Chat")
+    msg = gr.Textbox(placeholder="Type your message...", label="Your Message")
+    clear = gr.Button("🧹 Clear")
     state = gr.State([])
 
-    message.submit(chat_with_gemini, inputs=[message, state], outputs=[chatbot, state])
+    msg.submit(chat_with_gemini, inputs=[msg, state], outputs=[chatbot, state])
     clear.click(lambda: ([], []), None, [chatbot, state])
 
-# --- Launch app ---
+# --- Launch ---
 if __name__ == "__main__":
     demo.launch()