Update app.py

app.py

@@ -14,7 +14,6 @@ from dotenv import load_dotenv
 load_dotenv()
 
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
-genai.configure(api_key=GEMINI_API_KEY)
 
 TITLE = """<h1 align="center">🎮Chat with Gemini 1.5🔥 -- Beta Preview</h1>"""
 NOTICE = """
@@ -160,57 +159,64 @@ bot_inputs = [
     chatbot_component
 ]
 
-with gr.Blocks() as demo:
-    gr.HTML(TITLE)
-    with gr.Row():
-        gr.Markdown(NOTICE)
-        gr.Markdown(ABOUT)
-        gr.Markdown(ERRORS)
-        gr.Markdown(FUTURE_IMPLEMENTATIONS)
-    with gr.Column():
-        google_key_component.render()
+with gr.Blocks(theme = gr.Themes.Soft()) as demo:
+    with gr.Tab("Chat with Gemini 1.5 Flash"):
+        gr.HTML(TITLE)
         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()
+            gr.Markdown(NOTICE)
+            gr.Markdown(ABOUT)
+            gr.Markdown(ERRORS)
+            gr.Markdown(FUTURE_IMPLEMENTATIONS)
+        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
+            ]
+        )
+    with gr.Tab("Chat with Gemini Gemma 2"):
+        gr.HTML(
+            """
+            <h1 align="center">Still in development</h1>
+            """
+        )
+
+demo.queu().launch(debug = True, show_error = True)