Upload folder using huggingface_hub

.github/workflows/deploy-to-huggingface.yml

@@ -0,0 +1,28 @@
+name: Deploy to Hugging Face Space
+
+on:
+  push:
+    branches:
+      - main  # Trigger on pushes to the 'main' branch
+
+jobs:
+  build-and-deploy:
+    runs-on: ubuntu-latest  
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v3
+
+      - name: Setup Python
+        uses: actions/setup-python@v4
+        with:
+          python-version: '3.x'  
+
+      - name: Install dependencies
+        run: pip install -r requirements.txt
+
+      - name: Deploy to Hugging Face Space
+        uses: huggingface/space-deploy-action@v1
+        with:
+          repo_id: https://huggingface.co/spaces/airworkx/vertexchat  # Full URL
+          branch: main
+          build_script: 'python app.py'

.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__

.idea/.gitignore

@@ -0,0 +1,10 @@
+# Default ignored files
+/shelf/
+/workspace.xml
+# Editor-based HTTP Client requests
+/httpRequests/
+# Datasource local storage ignored files
+/dataSources/
+/dataSources.local.xml
+# Zeppelin ignored files
+/ZeppelinRemoteNotebooks/

.idea/inspectionProfiles/profiles_settings.xml

@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>

.idea/material_theme_project_new.xml

@@ -0,0 +1,10 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="MaterialThemeProjectNewConfig">
+    <option name="metadata">
+      <MTProjectMetadataState>
+        <option name="userId" value="-7b746248:18ffa6f84ed:-7ffd" />
+      </MTProjectMetadataState>
+    </option>
+  </component>
+</project>

.idea/misc.xml

@@ -0,0 +1,7 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="Black">
+    <option name="sdkName" value="Python 3.12 (vertexchat)" />
+  </component>
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.12 (vertexchat) (2)" project-jdk-type="Python SDK" />
+</project>

.idea/modules.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/vertexchat.iml" filepath="$PROJECT_DIR$/.idea/vertexchat.iml" />
+    </modules>
+  </component>
+</project>

.idea/vcs.xml

@@ -0,0 +1,7 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="" vcs="Git" />
+    <mapping directory="$PROJECT_DIR$/geminichat" vcs="Git" />
+  </component>
+</project>

.idea/vertexchat.iml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$" />
+    <orderEntry type="jdk" jdkName="Python 3.12 (vertexchat) (2)" jdkType="Python SDK" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>

.idea/workspace.xml

@@ -0,0 +1,185 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="AutoImportSettings">
+    <option name="autoReloadType" value="SELECTIVE" />
+  </component>
+  <component name="ChangeListManager">
+    <list default="true" id="8075f612-9024-458e-9030-3804cc940b48" name="Changes" comment="Removed .env file" />
+    <option name="SHOW_DIALOG" value="false" />
+    <option name="HIGHLIGHT_CONFLICTS" value="true" />
+    <option name="HIGHLIGHT_NON_ACTIVE_CHANGELIST" value="false" />
+    <option name="LAST_RESOLUTION" value="IGNORE" />
+  </component>
+  <component name="FileTemplateManagerImpl">
+    <option name="RECENT_TEMPLATES">
+      <list>
+        <option value="Python Script" />
+      </list>
+    </option>
+  </component>
+  <component name="Git.Settings">
+    <option name="RECENT_GIT_ROOT_PATH" value="$PROJECT_DIR$" />
+    <option name="RESET_MODE" value="HARD" />
+  </component>
+  <component name="GitHubPullRequestSearchHistory"><![CDATA[{
+  "lastFilter": {
+    "state": "OPEN",
+    "assignee": "omphcompany"
+  }
+}]]></component>
+  <component name="GithubPullRequestsUISettings"><![CDATA[{
+  "selectedUrlAndAccountId": {
+    "url": "https://github.com/omphcompany/vertexchat.git",
+    "accountId": "5b672839-567e-4290-a2d4-918066cd3c5c"
+  }
+}]]></component>
+  <component name="ProjectColorInfo"><![CDATA[{
+  "associatedIndex": 3
+}]]></component>
+  <component name="ProjectId" id="2hcVt4oWpHxV3A7vMpOEAm0Ls7w" />
+  <component name="ProjectLevelVcsManager">
+    <ConfirmationsSetting value="2" id="Add" />
+  </component>
+  <component name="ProjectViewState">
+    <option name="hideEmptyMiddlePackages" value="true" />
+    <option name="showLibraryContents" value="true" />
+  </component>
+  <component name="PropertiesComponent"><![CDATA[{
+  "keyToString": {
+    "ASKED_ADD_EXTERNAL_FILES": "true",
+    "RunOnceActivity.ShowReadmeOnStart": "true",
+    "com.google.cloudcode.ide_session_index": "20240608_0000",
+    "git-widget-placeholder": "master",
+    "last_opened_file_path": "/home/jackal/.virtualenvs/vertexchat",
+    "node.js.detected.package.eslint": "true",
+    "node.js.detected.package.tslint": "true",
+    "node.js.selected.package.eslint": "(autodetect)",
+    "node.js.selected.package.tslint": "(autodetect)",
+    "nodejs_package_manager_path": "npm",
+    "vue.rearranger.settings.migration": "true"
+  }
+}]]></component>
+  <component name="RdControllerToolWindowsLayoutState" isNewUi="true">
+    <layout>
+      <window_info id="Bookmarks" show_stripe_button="false" side_tool="true" />
+      <window_info id="Merge Requests" show_stripe_button="false" />
+      <window_info id="Commit_Guest" show_stripe_button="false" />
+      <window_info id="Learn" show_stripe_button="false" />
+      <window_info active="true" content_ui="combo" id="Project" order="0" sideWeight="0.5008038" visible="true" weight="0.32994792" />
+      <window_info id="Commit" order="1" sideWeight="0.5008038" weight="0.32994792" />
+      <window_info id="Pull Requests" order="2" sideWeight="0.5008038" weight="0.32994792" />
+      <window_info id="Welcome to GitHub Copilot" order="2" show_stripe_button="false" />
+      <window_info id="Structure" order="2" sideWeight="0.49919614" side_tool="true" weight="0.32994792" />
+      <window_info anchor="bottom" id="Database Changes" show_stripe_button="false" />
+      <window_info anchor="bottom" id="hadoop-toolwindow" show_stripe_button="false" />
+      <window_info anchor="bottom" id="KafkaToolWindow" show_stripe_button="false" />
+      <window_info anchor="bottom" id="EmrToolWindow" show_stripe_button="false" />
+      <window_info anchor="bottom" id="FlinkToolWindow" show_stripe_button="false" />
+      <window_info anchor="bottom" id="DataprocToolWindow" show_stripe_button="false" />
+      <window_info anchor="bottom" id="TypeScript" show_stripe_button="false" />
+      <window_info anchor="bottom" id="TODO" show_stripe_button="false" />
+      <window_info anchor="bottom" id="zeppelin-shell-toolwindow" show_stripe_button="false" />
+      <window_info anchor="bottom" id="File Transfer" show_stripe_button="false" />
+      <window_info anchor="bottom" id="HiveToolWindow" show_stripe_button="false" />
+      <window_info anchor="bottom" id="GlueToolWindow" show_stripe_button="false" />
+      <window_info anchor="bottom" id="Version Control" order="0" weight="0.32994187" />
+      <window_info anchor="bottom" id="Problems" order="1" />
+      <window_info anchor="bottom" id="Problems View" order="2" />
+      <window_info active="true" anchor="bottom" id="Terminal" order="3" visible="true" weight="0.32994187" />
+      <window_info anchor="bottom" id="Services" order="4" />
+      <window_info anchor="bottom" id="Python Packages" order="5" weight="0.32994187" />
+      <window_info anchor="bottom" id="Python Console" order="6" weight="0.32994187" />
+      <window_info anchor="right" id="Developer Tools" show_stripe_button="false" side_tool="true" />
+      <window_info anchor="right" id="Cloud Run" show_stripe_button="false" />
+      <window_info anchor="right" id="Compute Engine" show_stripe_button="false" />
+      <window_info anchor="right" id="SciView" show_stripe_button="false" />
+      <window_info anchor="right" id="Endpoints" show_stripe_button="false" />
+      <window_info anchor="right" id="Gemini: Chat" show_stripe_button="false" />
+      <window_info anchor="right" id="Source Protect" show_stripe_button="false" />
+      <window_info anchor="right" id="Kubernetes Explorer" show_stripe_button="false" />
+      <window_info anchor="right" id="BigDataToolWindow" show_stripe_button="false" />
+      <window_info anchor="right" id="Coverage" show_stripe_button="false" side_tool="true" />
+      <window_info anchor="right" id="Secret Manager" show_stripe_button="false" />
+      <window_info anchor="right" id="Google Cloud Storage (GCS)" show_stripe_button="false" />
+      <window_info anchor="right" id="Google Cloud Databases" show_stripe_button="false" />
+      <window_info anchor="right" content_ui="combo" id="Notifications" order="0" weight="0.25" />
+      <window_info active="true" anchor="right" id="AIAssistant" order="1" visible="true" weight="0.32994792" />
+      <window_info anchor="right" id="Database" order="2" weight="0.25" />
+      <window_info anchor="right" id="Gradle" order="3" weight="0.25" />
+      <window_info anchor="right" id="Maven" order="4" weight="0.25" />
+      <window_info anchor="right" id="github.copilotToolWindow" order="4" show_stripe_button="false" />
+      <window_info anchor="right" id="GitHub Copilot Chat" order="4" show_stripe_button="false" />
+      <window_info anchor="right" id="Github Tools" order="4" weight="0.32994792" />
+      <window_info anchor="right" id="Plots" order="5" weight="0.1" />
+    </layout>
+  </component>
+  <component name="SharedIndexes">
+    <attachedChunks>
+      <set>
+        <option value="bundled-js-predefined-1d06a55b98c1-0b3e54e931b4-JavaScript-PY-241.17011.84" />
+        <option value="bundled-python-sdk-afd503948556-7e9c3bbb6e34-com.jetbrains.pycharm.pro.sharedIndexes.bundled-PY-241.17011.84" />
+      </set>
+    </attachedChunks>
+  </component>
+  <component name="SpellCheckerSettings" RuntimeDictionaries="0" Folders="0" CustomDictionaries="0" DefaultDictionary="application-level" UseSingleDictionary="true" transferred="true" />
+  <component name="TaskManager">
+    <task active="true" id="Default" summary="Default task">
+      <changelist id="8075f612-9024-458e-9030-3804cc940b48" name="Changes" comment="" />
+      <created>1717893564913</created>
+      <option name="number" value="Default" />
+      <option name="presentableId" value="Default" />
+      <updated>1717893564913</updated>
+      <workItem from="1717893566223" duration="9130000" />
+    </task>
+    <task id="LOCAL-00001" summary="updated changes to gitignore">
+      <option name="closed" value="true" />
+      <created>1717900304672</created>
+      <option name="number" value="00001" />
+      <option name="presentableId" value="LOCAL-00001" />
+      <option name="project" value="LOCAL" />
+      <updated>1717900304672</updated>
+    </task>
+    <task id="LOCAL-00002" summary="Removed .env file">
+      <option name="closed" value="true" />
+      <created>1717902285080</created>
+      <option name="number" value="00002" />
+      <option name="presentableId" value="LOCAL-00002" />
+      <option name="project" value="LOCAL" />
+      <updated>1717902285080</updated>
+    </task>
+    <option name="localTasksCounter" value="3" />
+    <servers />
+  </component>
+  <component name="TypeScriptGeneratedFilesManager">
+    <option name="version" value="3" />
+  </component>
+  <component name="Vcs.Log.Tabs.Properties">
+    <option name="TAB_STATES">
+      <map>
+        <entry key="MAIN">
+          <value>
+            <State>
+              <option name="FILTERS">
+                <map>
+                  <entry key="branch">
+                    <value>
+                      <list>
+                        <option value="master" />
+                      </list>
+                    </value>
+                  </entry>
+                </map>
+              </option>
+            </State>
+          </value>
+        </entry>
+      </map>
+    </option>
+  </component>
+  <component name="VcsManagerConfiguration">
+    <option name="ADD_EXTERNAL_FILES_SILENTLY" value="true" />
+    <MESSAGE value="updated changes to gitignore" />
+    <MESSAGE value="Removed .env file" />
+    <option name="LAST_COMMIT_MESSAGE" value="Removed .env file" />
+  </component>
+</project>

.streamlit/config.toml

@@ -0,0 +1,2 @@
+[server]
+maxUploadSize=10

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_SETTINGS
+
+
+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_SETTINGS):
+                    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

1_Gemini_Pro_old.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

@@ -0,0 +1,17 @@
+
+## 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 app.py
+```
+Get your API KEY from https://makersuite.google.com
+
+
+# Credits:
+
+The repository was forked from: https://github.com/hiliuxg/geminichat

app.py

@@ -0,0 +1,74 @@
+import google.generativeai as genai
+import streamlit as st
+import time
+import random
+from utils import SAFETY_SETTINGS, MODIFIED_SAFETY_SETTINGS
+
+huggingface_token = os.getenv("HF_TOKEN")
+
+st.set_page_config(
+    page_title="Vertex Chat AI",
+    page_icon="🔥",
+    menu_items={
+        'About': "# Forked from https://github.com/omphcompany/geminichat"
+    }
+)
+
+st.title("Vertex Chat AI")
+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-1.5-flash-001')
+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_SETTINGS):
+                    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

apphf.py

@@ -0,0 +1,84 @@
+import google.generativeai as genai
+import streamlit as st
+import time
+import random
+import huggingface_hub
+from utils import SAFETY_SETTINGS
+
+st.set_page_config(
+    page_title="Vertex Chat AI",
+    page_icon="🔥",
+    menu_items={
+        'About': "# Forked from https://github.com/omphcompany/geminichat"
+    }
+)
+
+st.title("Vertex Chat AI")
+st.caption("Chatbot, powered by Google Gemini Pro.")
+
+# Hugging Face Hub Integration
+repo_id = "airworkx/vertexchat"  # Replace with your repo ID
+huggingface_token = st.text_input("Hugging Face Token", type='password')
+
+if "history" not in st.session_state:
+    st.session_state.history = []
+
+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
+
+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-1.5-flash-001')
+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 and huggingface_token:
+    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_SETTINGS):
+                    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)
+
+                # Save to Hugging Face Hub
+                huggingface_hub.login(token=huggingface_token)
+                huggingface_hub.create_repo(repo_id=repo_id, exist_ok=True)
+                with huggingface_hub.create_commit(repo_id=repo_id, revision="main", message="Added new conversation", commit_message="Added new conversation"):
+                    with open("conversation.txt", "a") as f:
+                        f.write(f"\n**User:** {prompt}\n**Assistant:** {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

apphf2.py

@@ -0,0 +1,113 @@
+import google.generativeai as genai
+import streamlit as st
+import time
+import random
+import huggingface_hub
+from utils import SAFETY_SETTINGS
+import json
+
+st.set_page_config(
+    page_title="Vertex Chat AI",
+    page_icon="🔥",
+    menu_items={
+        'About': "# Forked from https://github.com/omphcompany/geminichat"
+    }
+)
+
+st.title("Vertex Chat AI")
+st.caption("Chatbot, powered by Google Gemini Pro.")
+
+# Hugging Face Hub Integration
+repo_id = "airworkx/vertexchat"  # Your space ID
+huggingface_token = st.text_input("Hugging Face Token", type='password')
+
+if "history" not in st.session_state:
+    st.session_state.history = []
+
+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
+
+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-1.5-flash-001')
+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()
+
+# Load conversation history
+conversation_history = []
+try:
+    huggingface_hub.login(token=huggingface_token)
+    repo = huggingface_hub.Repo(repo_id=repo_id, revision="main")
+    with repo.file_obj("conversation_metadata.json", "r") as f:
+        conversation_history = json.load(f)
+except FileNotFoundError:
+    pass  # Initialize empty if file doesn't exist
+except ValueError:
+    st.warning("Invalid Hugging Face Token. Please check your token.")
+
+# Display the conversation history
+for entry in conversation_history:
+    if entry["role"] == "user":
+        with st.chat_message("user"):
+            st.markdown("[User Message]")
+    elif entry["role"] == "assistant":
+        with st.chat_message("assistant"):
+            st.markdown("[Assistant Message]")
+
+
+if "app_key" in st.session_state and huggingface_token:
+    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_SETTINGS):
+                    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)
+
+                # Update and save conversation history
+                conversation_history.append({"role": "user"})
+                conversation_history.append({"role": "assistant"})
+
+                # Save to Hugging Face Hub
+                try:
+                    huggingface_hub.login(token=huggingface_token)
+                    with open("conversation_metadata.json", "w") as f:
+                        json.dump(conversation_history, f)
+                    huggingface_hub.upload_file(
+                        path_or_fileobj="conversation_metadata.json",
+                        repo_id=repo_id,
+                        repo_type="space",  # Indicate it's a space
+                        revision="main",
+                        path_in_repo="conversation_metadata.json"
+                    )
+                except ValueError:
+                    st.warning("Invalid Hugging Face Token. Please check your token.")
+
+            except genai.types.generation_types.BlockedPromptException as e:
+                st.exception(e)
+            except Exception as e:
+                st.exception(e)

conversation_metadata.json

@@ -0,0 +1 @@
+[]

pages/2_Gemino_Pro_Vision.py

@@ -0,0 +1,86 @@
+from PIL import Image
+import google.generativeai as genai
+import streamlit as st
+import time
+import random
+from utils import SAFETY_SETTTINGS
+
+st.set_page_config(
+    page_title="Chat with Gemini-Pro Vision",
+    page_icon="🔥",
+    menu_items={
+        'About': "Forked from https://github.com/hiliuxg/geminichat"
+    }
+)
+
+st.title('Upload Image')
+
+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
+
+try:
+    genai.configure(api_key = st.session_state.app_key)
+    model = genai.GenerativeModel('gemini-pro-vision')
+except AttributeError as e:
+    st.warning("Please Add Your Gemini App Key.")
+
+
+def show_message(prompt, image, loading_str):
+    with st.chat_message("assistant"):
+        message_placeholder = st.empty()
+        message_placeholder.markdown(loading_str)
+        full_response = ""
+        try:
+            for chunk in model.generate_content([prompt, image], 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)
+        except genai.types.generation_types.BlockedPromptException as e:
+            st.exception(e)
+        except Exception as e:
+            st.exception(e)
+        message_placeholder.markdown(full_response)
+        st.session_state.history_pic.append({"role": "assistant", "text": full_response})
+
+def clear_state():
+    st.session_state.history_pic = []
+
+
+if "history_pic" not in st.session_state:
+    st.session_state.history_pic = []
+
+
+image = None
+if "app_key" in st.session_state:
+    uploaded_file = st.file_uploader("choose a pic...", type=["jpg", "png", "jpeg", "gif"], label_visibility='collapsed', on_change = clear_state)
+    if uploaded_file is not None:
+        image = Image.open(uploaded_file)
+        width, height = image.size
+        resized_img = image.resize((128, int(height/(width/128))), Image.LANCZOS)
+        st.image(image)    
+
+if len(st.session_state.history_pic) > 0:
+    for item in st.session_state.history_pic:
+        with st.chat_message(item["role"]):
+            st.markdown(item["text"])
+
+if "app_key" in st.session_state:
+    if prompt := st.chat_input("desc this picture"):
+        if image is None:
+            st.warning("Please upload an image first", icon="⚠️")
+        else:
+            prompt = prompt.replace('\n', '  \n')
+            with st.chat_message("user"):
+                st.markdown(prompt)
+                st.session_state.history_pic.append({"role": "user", "text": prompt})
+            
+            show_message(prompt, resized_img, "Thinking...")

pages/create_python_venv.sh

@@ -0,0 +1 @@
+    python3 -m venv env

requirements.txt

@@ -0,0 +1,4 @@
+streamlit==1.29.0
+google-generativeai==0.3.1
+huggingface-hub
+transformers

utils.py

@@ -0,0 +1,33 @@
+# Original code
+SAFETY_SETTINGS = [
+    {
+        "category": "HARM_CATEGORY_SEXUAL",
+        "threshold": "BLOCK_NONE",
+    },
+    {
+        "category": "HARM_CATEGORY_DANGEROUS",
+        "threshold": "BLOCK_NONE",
+    },
+    {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "threshold": "BLOCK_NONE",
+    },
+    {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "threshold": "BLOCK_NONE",
+    },
+    {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "threshold": "BLOCK_NONE",
+    },
+    {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "threshold": "BLOCK_NONE",
+    },
+]
+
+# Modified code to allow uninterrupted chat session
+MODIFIED_SAFETY_SETTTINGS = SAFETY_SETTINGS.copy()
+
+for setting in MODIFIED_SAFETY_SETTTINGS:
+    setting["threshold"] = "BLOCK_NONE" # Choose the appropriate threshold

utils_old.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",
+    },
+]