Update app.py

app.py

@@ -1,179 +1,136 @@
 import streamlit as st
 from g4f.client import Client
 import sqlite3
-import google.generativeai as genai
-# import pyttsx3
 import pyperclip
 import os
 
-api_key = os.getenv("Gemini_api")
+# --- KONFIGURASI GROQ (via environment variable) ---
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")  # Set di .env atau environment
+if not GROQ_API_KEY:
+    st.error("GROQ_API_KEY tidak ditemukan! Tambahkan di environment atau .env")
+    st.stop()
 
 
-
-
-st.set_page_config(page_title="DarkGPT",
-                   page_icon="🤖",
-                   initial_sidebar_state="expanded"
-)
-
+conn = sqlite3.connect('chat_history.db', check_same_thread=False)
+c = conn.cursor()
+c.execute('''CREATE TABLE IF NOT EXISTS chat_history
+             (conversation_id INTEGER, role TEXT, content TEXT)''')
+conn.commit()
 
 def local_css(file_name):
-    with open(file_name) as f:
-        st.markdown(f"<style>{f.read()}</style>", unsafe_allow_html=True)
-
+    try:
+        with open(file_name) as f:
+            st.markdown(f"<style>{f.read()}</style>", unsafe_allow_html=True)
+    except FileNotFoundError:
+        st.warning("style.css tidak ditemukan. Mengabaikan custom CSS.")
 
 local_css("style.css")
 
-# Create a connection to the database
-conn = sqlite3.connect('chat_history.db')
-c = conn.cursor()
+# --- MODEL LIST (via g4f + Groq) ---
+models = {
+    "🤖 Qwen 2.5 Coder (72B)": "qwen2.5-coder-32b",       
+    "🚀 LLaMA 3.1 70B": "llama-3.1-70b-versatile",
+    "⚡ LLaMA 3.1 8B": "llama-3.1-8b-instant",
+    "🧠 Qwen 2.5 72B": "qwen2.5-72b-instruct",
+    "⚙️ Mixtral 8x7B": "mixtral-8x7b-32768",
+}
 
-# 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}")
 
+client = Client(
+    provider="groq",
+    api_key=GROQ_API_KEY
+)
 
 
-# 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":
+    st.set_page_config(page_title="DarkGPT", page_icon="🤖", initial_sidebar_state="expanded")
+
+    # Inisialisasi session state
+    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
+
+    # Header + Model Selector
+    col1, col2, col3 = st.columns([1, 1, 1])
+    with col1:
+        st.header("DarkGPT")
+    with col3:
+        selected_model_name = st.selectbox("Pilih Model", list(models.keys()), index=0)
+        selected_model = models[selected_model_name]
+
+    # Sidebar: New Chat
+    if st.sidebar.button("✨ New Chat"):
+        st.session_state.chat_history = []
+        st.session_state.conversation_id += 1
+        st.rerun()
+
+    # Sidebar: Load History
+    st.sidebar.write("### Chat History")
+    c.execute("SELECT DISTINCT conversation_id FROM chat_history ORDER BY conversation_id DESC")
+    for (conv_id,) in c.fetchall():
+        c.execute("SELECT content FROM chat_history WHERE conversation_id = ? AND role = 'bot' LIMIT 1", (conv_id,))
+        first_msg = c.fetchone()
+        if first_msg:
+            preview = " ".join(first_msg[0].split()[:6]) + "..."
+            if st.sidebar.button(preview, key=f"load_{conv_id}"):
+                load_conversation(conv_id)
+
+    # Sidebar: Clear All
+    if st.sidebar.button("🗑️ Clear All History"):
+        c.execute("DELETE FROM chat_history")
+        conn.commit()
+        st.session_state.chat_history = []
+        st.rerun()
+
+    # Chat Input
+    st.markdown("---")
+    if user_input := st.chat_input("Ketik pesan..."):
+        with st.chat_message("user"):
+            st.markdown(user_input)
+
+        # Simpan user message
+        st.session_state.chat_history.append({"role": "user", "content": user_input})
+
+        # Generate response
+        with st.chat_message("assistant"):
+            with st.spinner("Thinking..."):
                 try:
-                    client = Client()
                     response = client.chat.completions.create(
-                        model=models[selected_model_display_name],
-                        messages=[{"role": "user", "content": user_input}],
+                        model=selected_model,
+                        messages=[{"role": m["role"], "content": m["content"]} for m in st.session_state.chat_history],
+                        temperature=0.7,
+                        max_tokens=2048
                     )
                     bot_response = response.choices[0].message.content
+                    st.markdown(bot_response)
 
-                    st.session_state.chat_history.append({"role": "user", "content": user_input})
-                    st.session_state.chat_history.append({"role": "bot", "content": bot_response})
+                    # Simpan bot response
+                    st.session_state.chat_history.append({"role": "assistant", "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"]))
+                    # Simpan ke DB
+                    for msg in st.session_state.chat_history[-2:]:  # hanya 2 pesan terakhir
+                        c.execute(
+                            "INSERT INTO chat_history VALUES (?, ?, ?)",
+                            (st.session_state.conversation_id, msg["role"], msg["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 = api_key
-                        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}")
-
-
-
+                    st.error(f"Error: {str(e)}")
 
 
-    except Exception as e:
-        st.error(f"An error occurred: {e}")
+    for msg in st.session_state.chat_history[:-2]:  
+        with st.chat_message(msg["role"]):
+            st.markdown(msg["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]}")
+def load_conversation(conv_id):
+    c.execute("SELECT role, content FROM chat_history WHERE conversation_id = ? ORDER BY rowid", (conv_id,))
+    st.session_state.chat_history = [{"role": r, "content": c} for r, c in c.fetchall()]
+    st.session_state.conversation_id = conv_id
+    st.rerun()
 
 
 if __name__ == "__main__":
-    main()
+    main()