Upload 5 files

.gitignore

@@ -0,0 +1,19 @@
+.venv
+.pytest_cache
+.mypy_cache
+.coverage
+.coverage.*
+.cache
+.cache.*
+.pytest_cache
+.pytest_cache.*
+.hypothesis
+.hypothesis.*
+.hypothesis_cache
+.hypothesis_cache.*
+.hypothesis_profile
+.hypothesis_profile.*
+.hypothesis_display
+.hypothesis_display.*
+.streamlit/secrets.toml
+__pycache__

1_Gemini_Pro.py

@@ -0,0 +1,75 @@
+
+
+import google.generativeai as genai
+import streamlit as st
+import time
+import random
+from utils import SAFETY_SETTTINGS
+
+
+st.set_page_config(
+    page_title="Gemini-Pro Chat",
+    page_icon="🔥",
+    menu_items={
+        'About': "# Forked from https://github.com/hiliuxg/geminichat"
+    }
+)
+
+st.title("Gemini-Pro Chat")
+st.caption("Chatbot, powered by Google Gemini Pro.")
+
+
+if "app_key" not in st.session_state:
+    app_key = st.text_input("Your Gemini App Key", type='password')
+    if app_key:
+        st.session_state.app_key = app_key
+
+if "history" not in st.session_state:
+    st.session_state.history = []
+
+try:
+    genai.configure(api_key = st.session_state.app_key)
+except AttributeError as e:
+    st.warning("Please Add Your Gemini App Key.")
+
+model = genai.GenerativeModel('gemini-pro')
+chat = model.start_chat(history = st.session_state.history)
+
+with st.sidebar:
+    if st.button("Clear Chat Window", use_container_width = True, type="primary"):
+        st.session_state.history = []
+        st.rerun()
+    
+for message in chat.history:
+    role = "assistant" if message.role == "model" else message.role
+    with st.chat_message(role):
+        st.markdown(message.parts[0].text)
+
+if "app_key" in st.session_state:
+    if prompt := st.chat_input(""):
+        prompt = prompt.replace('\n', '  \n')
+        with st.chat_message("user"):
+            st.markdown(prompt)
+
+        with st.chat_message("assistant"):
+            message_placeholder = st.empty()
+            message_placeholder.markdown("Thinking...")
+            try:
+                full_response = ""
+                for chunk in chat.send_message(prompt, stream=True, safety_settings = SAFETY_SETTTINGS):
+                    word_count = 0
+                    random_int = random.randint(5, 10)
+                    for word in chunk.text:
+                        full_response += word
+                        word_count += 1
+                        if word_count == random_int:
+                            time.sleep(0.05)
+                            message_placeholder.markdown(full_response + "_")
+                            word_count = 0
+                            random_int = random.randint(5, 10)
+                message_placeholder.markdown(full_response)
+            except genai.types.generation_types.BlockedPromptException as e:
+                st.exception(e)
+            except Exception as e:
+                st.exception(e)
+            st.session_state.history = chat.history

README.md

@@ -1,13 +1,17 @@
----
-title: Gemini
-emoji: 🚀
-colorFrom: blue
-colorTo: gray
-sdk: streamlit
-sdk_version: 1.30.0
-app_file: app.py
-pinned: false
-license: mit
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+## Gemini ChatBot
+Chatbot powered by Google Gemini-Pro, Gemini-Pro-Vision, and UI.
+
+https://geminix.streamlit.app/
+
+## Running Locally
+```python
+pip install -r requirements.txt
+streamlit run 1_Gemini_Pro.py
+```
+Get your API KEY from https://makersuite.google.com
+
+
+# Credits:
+
+The repository was forked from: https://github.com/hiliuxg/geminichat

requirements.txt

@@ -0,0 +1,2 @@
+streamlit==1.29.0
+google-generativeai==0.3.1

utils.py

@@ -0,0 +1,30 @@
+
+
+"""https://ai.google.dev/api/rest/v1beta/HarmCategory"""
+
+SAFETY_SETTTINGS = [
+    {
+        "category": "HARM_CATEGORY_SEXUAL",
+        "threshold": "BLOCK_ONLY_HIGH",
+    },
+    {
+        "category": "HARM_CATEGORY_DANGEROUS",
+        "threshold": "BLOCK_ONLY_HIGH",
+    },
+    {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "threshold": "BLOCK_ONLY_HIGH",
+    },
+    {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "threshold": "BLOCK_ONLY_HIGH",
+    },
+    {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "threshold": "BLOCK_ONLY_HIGH",
+    },
+    {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "threshold": "BLOCK_ONLY_HIGH",
+    },
+]