Update app.py

app.py

@@ -7,13 +7,11 @@ import gradio as gr
 from PIL import Image
 
 print("google-generativeai:", genai.__version__)
-
 GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
 
 # UI Titles and Subtitles
 TITLE = "<h1 align='center'>🚀 Gender Bias Detection App 🚀</h1>"
 SUBTITLE = "<h2 align='center'>Detect and analyze gender-based discrimination in communication.</h2>"
-
 IMAGE_WIDTH = 512
 
 def preprocess_stop_sequences(stop_sequences: str) -> Optional[List[str]]:
@@ -27,7 +25,6 @@ def user(text_prompt: str, chatbot: List[Tuple[str, str]]):
     return "", chatbot + [[text_prompt, None]]
 
 def bot(
-    google_key: str,
     image_prompt: Optional[Image.Image],
     temperature: float,
     max_output_tokens: int,
@@ -36,12 +33,11 @@ def bot(
     top_p: float,
     chatbot: List[Tuple[str, str]]
 ):
-    google_key = google_key or GOOGLE_API_KEY
-    if not google_key:
+    if not GOOGLE_API_KEY:
         raise ValueError("GOOGLE_API_KEY is not set. Please set it up.")
-
+    
     text_prompt = chatbot[-1][0]
-    genai.configure(api_key=google_key)
+    genai.configure(api_key=GOOGLE_API_KEY)  # Use the global API key
     generation_config = genai.types.GenerationConfig(
         temperature=temperature,
         max_output_tokens=max_output_tokens,
@@ -50,14 +46,14 @@ def bot(
         top_p=top_p,
         instructions="Analyze this text for gender-based discrimination, including implicit biases and stereotypes. Provide specific examples and explain why each example demonstrates bias.  Also, suggest tips for how to address or mitigate these biases within the text."
     )
-
-    model_name = "gemini-1.5-pro-latest" 
+    model_name = "gemini-1.5-pro-latest"
     model = genai.GenerativeModel(model_name)
+    
+    # Correctly handle inputs based on image_prompt
     inputs = [text_prompt] if image_prompt is None else [text_prompt, preprocess_image(image_prompt)]
     
     response = model.generate_content(inputs, stream=True, generation_config=generation_config)
     response.resolve()
-
     chatbot[-1][1] = ""
     for chunk in response:
         for i in range(0, len(chunk.text), 10):
@@ -74,11 +70,16 @@ with gr.Blocks() as demo:
         image_input = gr.Image(type="pil", label="Upload Image")
     submit_button = gr.Button("Analyze")
     chatbot_output = gr.Chatbot(label="Analysis Output")
-    
+
     submit_button.click(
+        fn=user,  # Call user function first
+        inputs=[text_input, chatbot_output],
+        outputs=[chatbot_output],
+        queue=False  # Prevent user input from being queued
+    ).then(
         fn=bot,
-        inputs=[GOOGLE_API_KEY, image_input, 0.4, 1024, "END", 32, 1, chatbot_output],
+        inputs=[image_input, 0.4, 1024, "END", 32, 1, chatbot_output],
         outputs=[chatbot_output]
     )
 
-demo.launch()
+demo.launch()