Added nightly version in app.py

app.py

@@ -15,6 +15,7 @@ load_dotenv()
 
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 
+# ============================== Stable - START ==============================
 TITLE = """<h1 align="center">🎮Chat with Gemini 1.5🔥 -- Beta Preview</h1>"""
 NOTICE = """
 Notices 📜:
@@ -155,8 +156,39 @@ bot_inputs = [
     top_p_component,
     chatbot_component
 ]
+# ============================== Stable - END ==============================
+
+# ============================== Nightly - START ==============================
+"""
+References:
+- https://medium.com/latinxinai/simple-chatbot-gradio-google-gemini-api-4ce02fbaf09f
+"""
+GEMINI_API_KEY_NIGHTLY = os.getenv("GEMINI_API_KEY_NIGHTLY")
+model_nightly_name = "gemini-1.5-flash"
+model_nightly = genai.GenerativeModel(model_nightly_name)
+chat = model_nightly.start_chat(history=[])
+
+def transform_history(history):
+    new_history = []
+    for chat in history:
+        new_history.append({"parts": [{"text": chat[0]}], "role": "user"})
+        new_history.append({"parts": [{"text": chat[1]}], "role": "model"})
+    return new_history
+
+def response(message, history):
+    global chat
+    chat.history = transform_history(history)
+    response = chat.send_message(message)
+    response.resolve()
+
+    for i in range(len(response.text)):
+        time.sleep(0.05)
+        yield response.text[:i+1]
+
+# ============================== Nightly - END ==============================
 
 with gr.Blocks(theme = gr.themes.Soft()) as demo:
+# ============================== Stable - START ==============================
     with gr.Tab("Chat with Gemini 1.5 Flash"):
         gr.HTML(TITLE)
         with gr.Row():
@@ -209,6 +241,7 @@ with gr.Blocks(theme = gr.themes.Soft()) as demo:
                 chatbot_component
             ]
         )
+# ============================== Stable - END ==============================
     
     with gr.Tab("Chat with Gemma 2"):
         gr.HTML(
@@ -217,11 +250,24 @@ with gr.Blocks(theme = gr.themes.Soft()) as demo:
             """
         )
 
+# ============================== Nightly - START ==============================
     with gr.Tab("Nightly -- Chat with Gemini 1.5"):
         gr.HTML(
             """
             <h1 align="center">This section will test out the next version of the stable version.</h1>
             """
         )
+        gr.ChatInterface(
+            response,
+            title = 'Gemini Chat',
+            textbox = gr.Textbox(
+                placeholder = "Chat with Gemini"
+            ),
+            retry_btn = None,
+            undo_btn = None,
+            clear_btn = None,
+            fill_height = True
+        )
+# ============================== Nightly - END ==============================
 
 demo.queue().launch(debug = True, show_error = True)