Spaces:
Running
Running
import streamlit as st | |
from g4f.client import Client | |
import sqlite3 | |
import google.generativeai as genai | |
# import pyttsx3 | |
import pyperclip | |
st.set_page_config(page_title="DarkGPT", | |
page_icon="๐ค", | |
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 = "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}") | |
except Exception as e: | |
st.error(f"An error occurred: {e}") | |
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() | |