Update app.py

app.py

@@ -10,11 +10,11 @@ print("google-generativeai:", genai.__version__)
 
 GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
 
-TITLE = """<h1 align="center">Gemini Playground 😎</h1>"""
-SUBTITLE = """<h2 align="center">Play with Gemini Pro and Gemini Pro Vision API 🖇️</h2>"""
+TITLE = """<h1 align="center">🕹️ Google Gemini Playground 🔥</h1>"""
+SUBTITLE = """<h2 align="center">Play with Gemini Text and Vision Model API 🖇️</h2>"""
 DUPLICATE = """
 <div style="text-align: center; display: flex; justify-content: center; align-items: center;">
-    <a href="https://huggingface.co/spaces/SkalskiP/ChatGemini?duplicate=true">
+    <a href="https://huggingface.co/spaces/ameerazam08/Gemini-All-in-One?logs=container&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 
@@ -44,6 +44,7 @@ def user(text_prompt: str, chatbot: List[Tuple[str, str]]):
 
 def bot(
     google_key: str,
+    model_name: str,
     image_prompt: Optional[Image.Image],
     temperature: float,
     max_output_tokens: int,
@@ -68,7 +69,7 @@ def bot(
         top_p=top_p)
 
     if image_prompt is None:
-        model = genai.GenerativeModel('gemini-pro')
+        model = genai.GenerativeModel(model_name)
         response = model.generate_content(
             text_prompt,
             stream=True,
@@ -103,6 +104,9 @@ google_key_component = gr.Textbox(
 )
 
 image_prompt_component = gr.Image(type="pil", label="Image", scale=1)
+
+model_selection = gr.Dropdown(["gemini-1.0-pro", "gemini-pro-vision","gemini-1.5-flash-latest", "gemini-1.5-pro-latest","gemini-1.0-pro-001"],label="Select Gemini Model",value="gemini-1.0-pro")
+
 chatbot_component = gr.Chatbot(
     label='Gemini',
     bubble_full_width=False,
@@ -178,6 +182,7 @@ user_inputs = [
 
 bot_inputs = [
     google_key_component,
+    model_selection, 
     image_prompt_component,
     temperature_component,
     max_output_tokens_component,
@@ -195,6 +200,7 @@ with gr.Blocks() as demo:
         google_key_component.render()
         with gr.Row():
             image_prompt_component.render()
+            model_selection.render()
             chatbot_component.render()
         text_prompt_component.render()
         run_button_component.render()