Update Nightly

Tabs/Gemini_Chabot_Nightly.py

@@ -1,3 +1,93 @@
+# import os
+# import google.generativeai as genai
+# from google.generativeai.types import HarmCategory, HarmBlockThreshold
+# import gradio as gr
+# from dotenv import load_dotenv
+
+# load_dotenv()
+
+# GEMINI_API_KEY_NIGHTLY = os.getenv("GEMINI_API_KEY_NIGHTLY")
+# model_name = "gemini-1.5-flash"
+
+# TITLE_NIGHTLY = """<h1 align="center">🎮Chat with Gemini 1.5🔥 -- Nightly</h1>"""
+# NOTICE_NIGHTLY = """
+# Notices 📜:
+# - This app is still in development (extreme unstable)
+# - Some features may not work as expected
+# - Currently the chatbot only support text and images
+# """
+# ERROR_NIGHTLY = """
+# Known errors ⚠️:
+# - Error when submit messages from uploading files.
+# """
+
+# def upload_to_gemini(path, mime_type=None):
+#     file = genai.upload_file(path, mime_type=mime_type)
+#     print(f"Uploaded file '{file.display_name}' as: {file.uri}")
+#     return file
+
+# def transform_history(history):
+#     new_history = []
+#     for user_msg, model_msg in history:
+#         new_history.append({"role": "user", "parts": [{"text": user_msg}]})
+#         new_history.append({"role": "model", "parts": [{"text": model_msg}]})
+#     return new_history
+
+# def chatbot_stable(message, history):
+#     message_text = message["text"]
+#     message_files = message["files"]
+#     print("Message text:", message_text)
+#     print("Message files:", message_files)
+#     if message_files:
+#         image_uris = [upload_to_gemini(file_path["path"]) for file_path in message_files]
+#         message_content = [{"text": message_text}] + image_uris
+#     else:
+#         message_content = {"text": message_text}
+#     genai.configure(api_key=GEMINI_API_KEY_NIGHTLY)
+#     model = genai.GenerativeModel(
+#         model_name,
+#         safety_settings={
+#             HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_ONLY_HIGH,
+#             HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_ONLY_HIGH,
+#             HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_ONLY_HIGH,
+#             HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_ONLY_HIGH,
+#         },
+#         generation_config={
+#             "temperature": 1,
+#             "top_p": 0.95,
+#             "top_k": 64,
+#             "max_output_tokens": 8192,
+#             "response_mime_type": "text/plain",
+#         }
+#     )
+
+#     global chat
+#     chat = model.start_chat(history=[])
+#     chat.history = transform_history(history)
+#     response = chat.send_message(message_content, stream=True)
+#     response.resolve()
+
+#     return response.text
+
+# gemini_chatbot_interface_stable = gr.Chatbot(
+#     height=500,
+#     likeable=True,
+#     avatar_images=(
+#         None,
+#         "https://media.roboflow.com/spaces/gemini-icon.png"
+#     ),
+#     show_copy_button=True,
+#     show_share_button=True,
+#     render_markdown=True
+# )
+
+# gemini_chatbot_stable = gr.ChatInterface(
+#     fn=chatbot_stable,
+#     chatbot=gemini_chatbot_interface_stable,
+#     title="Gemini 1.5 Chatbot",
+#     multimodal=True
+# )
+
 import os
 import google.generativeai as genai
 from google.generativeai.types import HarmCategory, HarmBlockThreshold
@@ -21,56 +111,53 @@ Known errors ⚠️:
 - Error when submit messages from uploading files.
 """
 
-def upload_to_gemini(path, mime_type=None):
-    file = genai.upload_file(path, mime_type=mime_type)
-    print(f"Uploaded file '{file.display_name}' as: {file.uri}")
-    return file
+genai.configure(api_key=GEMINI_API_KEY_NIGHTLY)
+model = genai.GenerativeModel(
+    model_name,
+    safety_settings={
+        HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_ONLY_HIGH,
+        HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_ONLY_HIGH,
+        HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_ONLY_HIGH,
+        HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_ONLY_HIGH,
+    },
+    generation_config={
+        "temperature": 1,
+        "top_p": 0.95,
+        "top_k": 64,
+        "max_output_tokens": 8192,
+        "response_mime_type": "text/plain",
+    }
+)
+chat = model.start_chat(history=[])
+
+def clear_chat_history():
+    chat.history = []
 
 def transform_history(history):
     new_history = []
     for user_msg, model_msg in history:
-        new_history.append({"role": "user", "parts": [{"text": user_msg}]})
-        new_history.append({"role": "model", "parts": [{"text": model_msg}]})
+        new_history.append({"role": "user", "parts": [user_msg]})
+        new_history.append({"role": "model", "parts": [model_msg]})
     return new_history
 
-def chatbot_stable(message, history):
+def chatbot_nightly(message, history):
     message_text = message["text"]
     message_files = message["files"]
     print("Message text:", message_text)
     print("Message files:", message_files)
     if message_files:
-        image_uris = [upload_to_gemini(file_path["path"]) for file_path in message_files]
-        message_content = [{"text": message_text}] + image_uris
+        image_uris = [genai.upload_file(path=file_path["path"]) for file_path in message_files]
+        message_content = [message_text] + image_uris
     else:
-        message_content = {"text": message_text}
-    genai.configure(api_key=GEMINI_API_KEY_NIGHTLY)
-    model = genai.GenerativeModel(
-        model_name,
-        safety_settings={
-            HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_ONLY_HIGH,
-            HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_ONLY_HIGH,
-            HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_ONLY_HIGH,
-            HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_ONLY_HIGH,
-        },
-        generation_config={
-            "temperature": 1,
-            "top_p": 0.95,
-            "top_k": 64,
-            "max_output_tokens": 8192,
-            "response_mime_type": "text/plain",
-        }
-    )
-
-    global chat
-    chat = model.start_chat(history=[])
-    chat.history = transform_history(history)
+        message_content = [message_text]
+
     response = chat.send_message(message_content, stream=True)
     response.resolve()
 
     return response.text
 
-gemini_chatbot_interface_stable = gr.Chatbot(
-    height=500,
+gemini_chatbot_interface_nightly = gr.Chatbot(
+    height=400,
     likeable=True,
     avatar_images=(
         None,
@@ -81,9 +168,15 @@ gemini_chatbot_interface_stable = gr.Chatbot(
     render_markdown=True
 )
 
-gemini_chatbot_stable = gr.ChatInterface(
-    fn=chatbot_stable,
-    chatbot=gemini_chatbot_interface_stable,
+clear_chat_button = gr.ClearButton(
+    components=[gemini_chatbot_interface_nightly],
+    value="🗑️ Clear"
+)
+
+gemini_chatbot_nightly = gr.ChatInterface(
+    fn=chatbot_nightly,
+    chatbot=gemini_chatbot_interface_nightly,
     title="Gemini 1.5 Chatbot",
-    multimodal=True
+    multimodal=True,
+    clear_btn=clear_chat_button
 )

app.py

@@ -11,7 +11,7 @@ import gradio as gr
 from PIL import Image
 from dotenv import load_dotenv
 from google.generativeai.types import HarmCategory, HarmBlockThreshold
-from Tabs.Gemini_Chabot_Nightly import TITLE_NIGHTLY, NOTICE_NIGHTLY, ERROR_NIGHTLY, gemini_chatbot_stable
+from Tabs.Gemini_Chabot_Nightly import gemini_chatbot_nightly, clear_chat_button, clear_chat_history, TITLE_NIGHTLY, NOTICE_NIGHTLY, ERROR_NIGHTLY
 
 load_dotenv()
 
@@ -225,7 +225,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         with gr.Row():
             gr.Markdown(NOTICE_NIGHTLY)
             gr.Markdown(ERROR_NIGHTLY)
-        gemini_chatbot_stable.render()
+        gemini_chatbot_nightly.render()
+        clear_chat_button.click(
+            fn=clear_chat_history
+        )
 # ============================== Nightly - END ==============================
 
 demo.queue().launch(debug=True, show_error=True)