initial commit

.github/workflows/sync_to_huggingface_hub.yml

@@ -0,0 +1,20 @@
+name: Sync to HuggingFace hub
+on:
+  push:
+    branches: [main]
+
+  # to run this workflow manually from the Actions tab
+  workflow_dispatch:
+
+jobs:
+  sync-to-hub:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v3
+        with:
+          fetch-depth: 0
+          lfs: true
+      - name: Push to hub
+        env:
+          HF_TOKEN: ${{ secrets.HF_TOKEN }}
+        run: git push --force https://thivav:$HF_TOKEN@huggingface.co/spaces/thivav/chat_with_gemini main

.gitignore

@@ -157,4 +157,6 @@ cython_debug/
 #  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
-#.idea/
+.idea/
+
+vscode/

README.md

@@ -1,2 +1,12 @@
-# gemini_conversational_chatbot
-Google Gemini Chatbot
+title: Chat With Gemini
+emoji: 🤖
+colorFrom: pink
+colorTo: pink
+sdk: streamlit
+sdk_version: 1.31.0
+app_file: app.py
+pinned: false
+
+# Chatbot with Gemini
+
+[Google Gemini Chatbot - Playground](https://huggingface.co/spaces/thivav/chat_with_gemini)

app.py

@@ -0,0 +1,58 @@
+import google.generativeai as genai
+import streamlit as st
+
+
+def main():
+    """main"""
+
+    st.set_page_config(
+        page_title="Gemini Chatbot",
+        page_icon="🤖",
+        layout="centered",
+        initial_sidebar_state="collapsed",
+    )
+
+    st.header("The Gemini Chatbot 🤖", divider="rainbow")
+
+    st.subheader(
+        "Enjoy :red[talking] with :green[Google Gemini] :sunglasses:"
+    )
+
+    st.markdown("[check out the repository](https://github.com/ThivaV/gemini_chatbot)")
+
+    gemini_key = st.text_input("Enter your Google Gemini API key 👇", type="password")
+
+    genai.configure(api_key=gemini_key)
+
+    model = genai.GenerativeModel("gemini-pro")
+    chat = model.start_chat(history=[])
+
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"]):
+            st.markdown(message["content"])
+
+    if prompt := st.chat_input("Say something"):
+        # user message
+        with st.chat_message("user"):
+            st.markdown(prompt)
+            st.session_state.messages.append({"role": "user", "content": prompt})
+
+        response = chat.send_message(prompt, stream=True)
+        complete_response = ""
+        with st.chat_message("assistant"):
+            for chunk in response:
+                complete_response += chunk.text
+                st.write(chunk.text)
+
+            st.session_state.messages.append(
+                {"role": "assistant", "content": complete_response}
+            )
+
+
+if __name__ == "__main__":
+
+    # initialize streamlit session state
+    if "messages" not in st.session_state:
+        st.session_state.messages = []
+
+    main()

requirements.txt

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