Update app.py

app.py

@@ -7,11 +7,22 @@ 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>"
+TITLE = """<h1 align="center">🕹️ Google Gemini Chatbot 🔥</h1>"""
+SUBTITLE = """<h2 align="center">🎨Create with Multimodal Gemini</h2>"""
+DUPLICATE = """
+<div style="text-align: center; display: flex; justify-content: center; align-items: center;">
+    <a href="https://huggingface.co/spaces/Rahatara/build_with_gemini/blob/main/allgemapp.py?duplicate=true">
+        <img src="https://bit.ly/3gLdBN6" alt="Duplicate Space" style="margin-right: 10px;">
+    </a>
+    <span>Duplicate the Space and run securely with your 
+        <a href="https://makersuite.google.com/app/apikey">GOOGLE API KEY</a>.
+    </span>
+</div>
+"""
+
 IMAGE_WIDTH = 512
 
 def preprocess_stop_sequences(stop_sequences: str) -> Optional[List[str]]:
@@ -25,6 +36,7 @@ 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,
@@ -33,27 +45,28 @@ def bot(
     top_p: float,
     chatbot: List[Tuple[str, str]]
 ):
-    if not GOOGLE_API_KEY:
+    google_key = google_key or GOOGLE_API_KEY
+    if not google_key:
         raise ValueError("GOOGLE_API_KEY is not set. Please set it up.")
-    
+
     text_prompt = chatbot[-1][0]
-    genai.configure(api_key=GOOGLE_API_KEY)  # Use the global API key
+    genai.configure(api_key=google_key)
     generation_config = genai.types.GenerationConfig(
         temperature=temperature,
         max_output_tokens=max_output_tokens,
         stop_sequences=preprocess_stop_sequences(stop_sequences),
         top_k=top_k,
         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."
+        instructions = "You are an advocate against gender-based discrimination "
     )
-    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):
@@ -61,15 +74,45 @@ def bot(
             time.sleep(0.01)
             yield chatbot
 
-# Gradio Interface
+google_key_component = gr.Textbox(
+    label="GOOGLE API KEY",
+    type="password",
+    placeholder="...",
+    visible=GOOGLE_API_KEY is None
+)
+
+image_prompt_component = gr.Image(type="pil", label="Image")
+chatbot_component = gr.Chatbot(label='Gemini', bubble_full_width=False)
+text_prompt_component = "Analyze this for any instances of gender-based discrimination. Consider both explicit and implicit biases, stereotypes, and unequal treatment. Provide specific examples from the text to support your analysis" 
+run_button_component = gr.Button("Check Discrimantion")
+temperature_component = gr.Slider(minimum=0, maximum=1.0, value=0.4, step=0.05, label="Temperature")
+max_output_tokens_component = gr.Slider(minimum=1, maximum=2048, value=1024, step=1, label="Token limit")
+stop_sequences_component = gr.Textbox(label="Add stop sequence", placeholder="STOP, END")
+top_k_component = gr.Slider(minimum=1, maximum=40, value=32, step=1, label="Top-K")
+top_p_component = gr.Slider(minimum=0, maximum=1, value=1, step=0.01, label="Top-P")
+
+user_inputs = [text_prompt_component, chatbot_component]
+bot_inputs = [google_key_component, image_prompt_component, temperature_component, max_output_tokens_component, stop_sequences_component, top_k_component, top_p_component, chatbot_component]
+
 with gr.Blocks() as demo:
-    gr.Markdown(TITLE)
-    gr.Markdown(SUBTITLE)
-    with gr.Row():
-        text_input = gr.Textbox(placeholder="Enter text to analyze for gender-based discrimination")
-        image_input = gr.Image(type="pil", label="Upload Image")
-    submit_button = gr.Button("Analyze")
-    chatbot_output = gr.Chatbot(label="Analysis Output")
-    submit_btn.click(fn=respond, inputs=[txt_input, chatbot], outputs=[chatbot, txt_input])
-
-demo.launch()
+    gr.HTML(TITLE)
+    gr.HTML(SUBTITLE)
+    gr.HTML(DUPLICATE)
+    with gr.Column():
+        google_key_component.render()
+        with gr.Row():
+            image_prompt_component.render()
+            chatbot_component.render()
+        text_prompt_component.render()
+        run_button_component.render()
+        with gr.Accordion("Parameters", open=False):
+            temperature_component.render()
+            max_output_tokens_component.render()
+            stop_sequences_component.render()
+            with gr.Accordion("Advanced", open=False):
+                top_k_component.render()
+                top_p_component.render()
+
+    run_button_component.click(fn=user, inputs=user_inputs, outputs=[text_prompt_component, chatbot_component], queue=False).then(fn=bot, inputs=bot_inputs, outputs=[chatbot_component])
+    text_prompt_component.submit(fn=user, inputs=user_inputs, outputs=[text_prompt_component, chatbot_component], queue=False).then(fn=bot, inputs=bot_inputs, outputs=[chatbot_component])
+demo.launch()