Update app.py

app.py

@@ -1,61 +1,33 @@
-# Importing necessary Python libraries
-import os  # Access environment variables like secret keys
-import gradio as gr  # Build simple web interfaces
-import google.generativeai as genai  # Connect to Google Gemini API
+import os
+import gradio as gr
+import google.generativeai as genai
 
-# -------------------------
-# 🛠️ Gemini API Setup
-# -------------------------
+GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
 
-# Get the Google API key from environment (stored securely in Hugging Face Secrets)
-GOOGLE_API_KEY = oos.environ["GOOGLE_API_KEY"]
+if not GOOGLE_API_KEY:
+    raise ValueError("GOOGLE_API_KEY is missing. Add it in HF Secrets.")
 
-# Configure Gemini API
 genai.configure(api_key=GOOGLE_API_KEY)
 
-# -------------------------
-# 💬 Chat Function
-# -------------------------
-
-# Main function for conversation with Gemini
 def chat_with_gemini(history, message):
-    # Convert Gradio’s history to a simple prompt for Gemini
     conversation = ""
     for user_msg, bot_msg in history:
-        conversation += f"User: {user_msg}\n"
-        if bot_msg:
-            conversation += f"Assistant: {bot_msg}\n"
-    
-    # Add the new user message
+        conversation += f"User: {user_msg}\nAssistant: {bot_msg}\n"
+
     conversation += f"User: {message}\nAssistant:"
 
     try:
-        # Use Gemini model to generate response
-        model = genai.GenerativeModel('gemini-1.5-flash')
+        model = genai.GenerativeModel("gemini-1.5-flash")
         response = model.generate_content(conversation)
         bot_reply = response.text.strip()
     except Exception as e:
-        # Handle errors
-        bot_reply = f"⚠️ Error: {str(e)}"
+        bot_reply = f"⚠️ Error: {e}"
 
-    # Add to history
     history.append((message, bot_reply))
-
-    # Return for Gradio
     return history, history
 
-# -------------------------
-# 🖼️ Gradio Chat Interface (UI)
-# -------------------------
-
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown(
-        """
-        # 🤖 EchoForge Chatbot
-        Chat with Google's Gemini AI! Type below to start.
-        """
-    )
-    
+    gr.Markdown("# 🤖 EchoForge Chatbot")
     chatbot = gr.Chatbot([], height=400)
     msg = gr.Textbox(placeholder="Type your message...")
     clear = gr.Button("Clear Chat")
@@ -63,9 +35,4 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     msg.submit(chat_with_gemini, [chatbot, msg], [chatbot, chatbot])
     clear.click(lambda: [], outputs=[chatbot])
 
-# -------------------------
-# 🚀 Launch the App
-# -------------------------
-
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()