Create DarkGPT.py

Author: codewithdark-git

pages/DarkGPT.py

@@ -0,0 +1,179 @@
+import streamlit as st
+from g4f.client import Client
+import sqlite3
+import google.generativeai as genai
+import csv
+import os
+# import pyttsx3
+import pyperclip
+
+
+st.set_page_config(page_title="DarkGPT",
+                   page_icon="🤖",
+                   layout="wide",
+                   initial_sidebar_state="expanded"
+)
+def local_css(file_name):
+    with open(file_name) as f:
+        st.markdown(f"<style>{f.read()}</style>", unsafe_allow_html=True)
+
+
+local_css("style.css")
+
+# Create a connection to the database
+conn = sqlite3.connect('chat_history.db')
+c = conn.cursor()
+
+# Create table if not exists
+try:
+    c.execute('''CREATE TABLE IF NOT EXISTS chat_history
+                 (conversation_id INTEGER, role TEXT, content TEXT)''')
+    conn.commit()
+except Exception as e:
+    st.error(f"An error occurred: {e}")
+
+# Streamlit app
+def main():
+    try:
+        if "chat_history" not in st.session_state:
+            st.session_state.chat_history = []
+
+        if "conversation_id" not in st.session_state:
+            st.session_state.conversation_id = 1
+
+        models = {
+            "🚀 Airoboros 70B": "airoboros-70b",
+            "👑 Gemini 1.0": "gemini-1.0-pro",
+            "🧨 Gemini 1.0 Pro ": "gemini-1.0-pro-001",
+            "⚡ Gemini 1.0 pro latest": "gemini-1.0-pro-latest",
+            "🔮 Gemini Pro": "gemini-pro"
+        }
+
+        columns = st.columns(3)  # Split the layout into three columns
+        with columns[0]:
+            st.header("DarkGPT")
+
+        with columns[2]:
+
+            selected_model_display_name = st.selectbox("Select Model", list(models.keys()), index=0)
+            selected_model = models[selected_model_display_name]
+
+        with columns[1]:
+            pass
+            # if st.button("summarize"):
+            #     st.switch_page('pages/summarize.py')
+
+        # Sidebar (left side) - New chat button
+        if st.sidebar.button("✨ New Chat", key="new_chat_button"):
+            st.session_state.chat_history.clear()
+            st.session_state.conversation_id += 1
+
+        # Sidebar (left side) - Display saved chat
+        st.sidebar.write("Chat History")
+        c.execute("SELECT DISTINCT conversation_id FROM chat_history")
+        conversations = c.fetchall()
+        for conv_id in reversed(conversations):
+            c.execute("SELECT content FROM chat_history WHERE conversation_id=? AND role='bot' LIMIT 1",
+                      (conv_id[0],))
+            first_bot_response = c.fetchone()
+            if first_bot_response:
+                if st.sidebar.button(" ".join(first_bot_response[0].split()[0:5])):
+                    display_conversation(conv_id[0])
+
+        # Sidebar (left side) - Clear Chat History button
+        if st.sidebar.button("Clear Chat History ✖️"):
+            st.session_state.chat_history.clear()
+            c.execute("DELETE FROM chat_history")
+            conn.commit()
+
+        # Main content area (center)
+        st.markdown("---")
+
+        user_input = st.chat_input("Ask Anything ...")
+
+        if user_input:
+            if selected_model == "airoboros-70b":
+                try:
+                    client = Client()
+                    response = client.chat.completions.create(
+                        model=models[selected_model_display_name],
+                        messages=[{"role": "user", "content": user_input}],
+                    )
+                    bot_response = response.choices[0].message.content
+
+                    st.session_state.chat_history.append({"role": "user", "content": user_input})
+                    st.session_state.chat_history.append({"role": "bot", "content": bot_response})
+
+                    # Store chat in the database
+                    for chat in st.session_state.chat_history:
+                        c.execute("INSERT INTO chat_history VALUES (?, ?, ?)",
+                                  (st.session_state.conversation_id, chat["role"], chat["content"]))
+                    conn.commit()
+
+                    # Display chat history
+                    for index, chat in enumerate(st.session_state.chat_history):
+                        with st.chat_message(chat["role"]):
+                            if chat["role"] == "user":
+                                st.markdown(chat["content"])
+                            elif chat["role"] == "bot":
+                                st.markdown(chat["content"])
+
+                except Exception as e:
+                    st.error(f"An error occurred: {e}")
+
+            else:
+                try:
+                        # GEMINI Replace with your Gemini Api key
+                        GOOGLE_API_KEY = os.getenv('gemini-api')
+                        genai.configure(api_key=GOOGLE_API_KEY)
+                        model = genai.GenerativeModel(selected_model)
+                        prompt = user_input
+                        response = model.generate_content(prompt)
+                        bot_response = response.candidates[0].content.parts[0].text
+
+                        st.session_state.chat_history.append({"role": "user", "content": user_input})
+                        st.session_state.chat_history.append({"role": "bot", "content": bot_response})
+
+                        # Store chat in the database
+                        for chat in st.session_state.chat_history:
+                            c.execute("INSERT INTO chat_history VALUES (?, ?, ?)",
+                                      (st.session_state.conversation_id, chat["role"], chat["content"]))
+                        conn.commit()
+
+                        for index, chat in enumerate(st.session_state.chat_history):
+                            with st.chat_message(chat["role"]):
+                                if chat["role"] == "user":
+                                    st.markdown(chat["content"])
+                                elif chat["role"] == "bot":
+                                    st.markdown(chat["content"])
+
+                except Exception as e:
+                     st.error(f"An error occurred: {e}")
+
+            # export_to_csv(st.session_state.chat_history)
+
+
+    except Exception as e:
+        st.error(f"An error occurred: {e}")
+
+# def export_to_csv(chat_history):
+#     filename = "chat_history.csv"
+#     latest_conversation = chat_history[-2:]  # Get only the latest conversation
+#     with open(filename, "a+", newline="") as csvfile:
+#         fieldnames = ["User Input", "Bot Response"]
+#         writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
+#         if csvfile.tell() == 0:  # Check if the file is empty
+#             writer.writeheader()  # Write header if file is empty
+#         if len(latest_conversation) == 2:  # Check if the latest conversation is complete
+#             writer.writerow({"User Input": latest_conversation[0]["content"], "Bot Response": latest_conversation[1]["content"]})
+
+def display_conversation(conversation_id):
+    c.execute("SELECT * FROM chat_history WHERE conversation_id=?", (conversation_id,))
+    chats = c.fetchall()
+    st.markdown(f"### Conversation")
+    for chat in chats:
+        st.markdown(f"{chat[1]}")
+        st.markdown(f"{chat[2]}")
+
+if __name__ == "__main__":
+    main()