Update app.py

app.py

@@ -2,25 +2,35 @@ import gradio as gr
 import google.generativeai as genai
 import os
 
-# Configure the Gemini API with your API key
-genai.configure(api_key=os.environ["YOUR_API_KEY"])
-
-# Initialize the Gemini Generative Model
-model = genai.GenerativeModel("gemini-1.5-flash")
-
-def chat_with_gemini(user_input, history):
+def chat_with_gemini(user_api_key, user_input, history):
     """
-    Generates a response from the Gemini API based on user input and conversation history.
+    Generates a response from the Gemini API based on user input and conversation history,
+    using the provided user API key or falling back on a default API key.
     
     Args:
+        user_api_key (str): The user's Google Gemini API key (optional).
         user_input (str): The latest message from the user.
         history (list): The conversation history as a list of tuples.
     
     Returns:
-        tuple: The chatbot's reply and the updated history.
+        tuple: The updated conversation history with the chatbot's reply.
     """
+    # Determine which API key to use
+    api_key = user_api_key.strip() if user_api_key else os.getenv("DEFAULT_API_KEY")
+    
+    if not api_key:
+        # If no API key is available, prompt the user
+        history.append(("Bot", "Please enter your Google Gemini API key to start the conversation."))
+        return history, history
+    
     try:
-        # Send the latest message to the Gemini API
+        # Configure the Gemini API with the selected API key
+        genai.configure(api_key=api_key)
+        
+        # Initialize the Gemini Generative Model
+        model = genai.GenerativeModel("gemini-1.5-flash")
+        
+        # Generate a response from the Gemini API
         response = model.generate_content(
             user_input,
             generation_config=genai.GenerationConfig(
@@ -32,47 +42,71 @@ def chat_with_gemini(user_input, history):
         chatbot_reply = response.text.strip()
         
         # Append the user input and chatbot reply to the history
-        history.append((user_input, chatbot_reply))
+        history.append(("User", user_input))
+        history.append(("Bot", chatbot_reply))
         
         return history, history
     except Exception as e:
         error_message = f"An error occurred: {e}"
-        history.append((user_input, error_message))
+        history.append(("Bot", error_message))
         return history, history
 
 with gr.Blocks() as iface:
-    gr.Markdown("# Google Gemini 1.5 Flash Chatbot")
+    gr.Markdown("""
+    # Google Gemini Flash 1.5 Chatbot
+    
+    Welcome to the Google Gemini-powered chatbot! You can interact with the bot by typing your messages below.
+    
+    **API Key Setup:**
+    - **Option 1:** Enter your own Google Gemini API key in the input field below.
+    - **Option 2:** If you leave the API key field empty, the chatbot will use a default API key.
     
-    chatbot = gr.Chatbot()
+    > **Note:** Ensure that your API key is kept secure and do not share it publicly.
+    """)
     
-    with gr.Row():
-        with gr.Column(scale=0.85):
+    with gr.Column():
+        # API Key Input Section
+        with gr.Row():
+            api_key_input = gr.Textbox(
+                label="Google Gemini API Key (Optional)",
+                placeholder="Enter your API key here...",
+                type="password",
+                lines=1
+            )
+        
+        # Chatbot Display
+        chatbot = gr.Chatbot()
+        
+        # User Input Row
+        with gr.Row():
             user_input = gr.Textbox(
                 placeholder="Type your message here...",
                 show_label=False
             )
-        with gr.Column(scale=0.15):
             send_button = gr.Button("Send")
     
     # State to hold the conversation history
     history = gr.State([])
     
-    def respond(message, history_state):
+    def respond(user_api_key, message, history_state):
         """
-        Handles the user message, generates a response, and updates the conversation history.
+        Handles the user message, generates a response using the provided or default API key,
+        and updates the conversation history.
         
         Args:
+            user_api_key (str): The user's API key (optional).
             message (str): The user's message.
             history_state (list): The current conversation history.
         
         Returns:
             tuple: Updated conversation history for display.
         """
-        updated_history, new_history = chat_with_gemini(message, history_state)
+        updated_history, new_history = chat_with_gemini(user_api_key, message, history_state)
         return updated_history, new_history
     
-    send_button.click(respond, inputs=[user_input, history], outputs=[chatbot, history])
-    user_input.submit(respond, inputs=[user_input, history], outputs=[chatbot, history])
+    # Connect the send button and textbox submission to the respond function
+    send_button.click(respond, inputs=[api_key_input, user_input, history], outputs=[chatbot, history])
+    user_input.submit(respond, inputs=[api_key_input, user_input, history], outputs=[chatbot, history])
 
 if __name__ == "__main__":
     iface.launch()