Update app.py

app.py

@@ -5,7 +5,7 @@ import google.generativeai as genai
 import gradio as gr
 
 # Get the Google API key from environment variables
-google_key = os.environ.get("GOOGLE_API_KEY")
+GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
 
 # Define the title and subtitle for the Gradio interface
 TITLE = """<h1 align="center">🏋️ AI Personal Trainer Playground 💪</h1>"""
@@ -34,6 +34,7 @@ def user(text_prompt: str, chatbot: List[Tuple[str, str]]):
 
 # Function to handle bot response
 def bot(
+    google_key: str,
     model_name: str,
     video_prompt,
     temperature: float,
@@ -44,7 +45,13 @@ def bot(
     text_prompt_component: str,
     chatbot: List[Tuple[str, str]]
 ):
-    
+    # Use the provided Google API key or the one from environment variables
+    google_key = google_key if google_key else GOOGLE_API_KEY
+    if not google_key:
+        raise ValueError(
+            "GOOGLE_API_KEY is not set. "
+            "Please follow the instructions in the README to set it up.")
+
     # Combine the user input with the predefined prompt
     user_input = chatbot[-1][0]
     combined_prompt = Prompt + "\n" + user_input
@@ -96,6 +103,15 @@ def bot(
             time.sleep(0.01)
             yield chatbot
 
+# Define Gradio components
+google_key_component = gr.Textbox(
+    label="GOOGLE API KEY",
+    value="",
+    type="password",
+    placeholder="...",
+    info="You have to provide your own GOOGLE_API_KEY for this app to function properly",
+    visible=GOOGLE_API_KEY is None
+)
 
 video_prompt_component = gr.Video(label="Video", autoplay=True)
 
@@ -173,6 +189,7 @@ top_p_component = gr.Slider(
 user_inputs = [text_prompt_component, chatbot_component]
 
 bot_inputs = [
+    google_key_component,
     model_selection, 
     video_prompt_component,
     temperature_component,
@@ -189,6 +206,7 @@ with gr.Blocks() as demo:
     gr.HTML(TITLE)
     gr.HTML(SUBTITLE)
     with gr.Column():
+        google_key_component.render()
         with gr.Row():
             video_prompt_component.render()
             chatbot_component.render()