Update app.py

app.py

@@ -1,5 +1,5 @@
 import os
-import openai
+from groq import Groq
 import streamlit as st
 from dotenv import load_dotenv
 from typing import List
@@ -8,12 +8,11 @@ import abc
 
 # Load environment variables (API keys)
 load_dotenv()
-openai.api_key = os.getenv("OPENAI_API_KEY")
 
 # Define constants
 CONVO_TRAIL_CUTOFF = 5
 PERSONAL_AI_ASSISTANT_PROMPT_HEAD = "You are a helpful assistant. [[previous_interactions]] [[latest_input]]"
-ASSISTANT_TYPE = "OpenAIPAF"
+ASSISTANT_TYPE = "GroqPAF"
 
 # Define Interaction class
 class Interaction:
@@ -38,21 +37,27 @@ class PersonalAssistantFramework(abc.ABC):
     def think(self, prompt: str) -> str:
         pass
 
-class OpenAIPAF(PersonalAssistantFramework):
+class GroqPAF(PersonalAssistantFramework):
+    def __init__(self):
+        self.client = Groq(api_key=os.getenv("GROQ_API_KEY"))
+
     def setup(self):
-        self.llm_model = None  # Assume build_mini_model is not defined
+        self.llm_model = None
 
     @PersonalAssistantFramework.timeit_decorator
     def think(self, thought: str) -> str:
-        response = openai.ChatCompletion.create(
-            model="gpt-4o-mini",
-            messages=[
-                {"role": "system", "content": "You are a helpful assistant."},
-                {"role": "user", "content": thought}
-            ],
-            max_tokens=500
-        )
-        return response.choices[0].message['content'].strip()
+        try:
+            response = self.client.chat.completions.create(
+                model="mixtral-8x7b-32768",  # You can change this to other Groq models
+                messages=[
+                    {"role": "system", "content": "You are a helpful assistant named Luna."},
+                    {"role": "user", "content": thought}
+                ],
+                max_tokens=500
+            )
+            return response.choices[0].message.content.strip()
+        except Exception as e:
+            return f"An error occurred: {str(e)}"
 
 def build_prompt(latest_input: str, previous_interactions: List[Interaction]) -> str:
     previous_interactions_str = "\n".join(
@@ -71,7 +76,7 @@ def build_prompt(latest_input: str, previous_interactions: List[Interaction]) ->
     return prepared_prompt
 
 # Set the page config for Streamlit
-st.set_page_config(page_title="AI Chatbot", page_icon="🤖", layout="wide")
+st.set_page_config(page_title="Luna AI Chatbot", page_icon="🤖", layout="wide")
 
 # Add premium CSS styling with the new blue color palette
 st.markdown("""
@@ -81,7 +86,6 @@ st.markdown("""
         color: #AFDDE5;             /* Light blue text for contrast */
         font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif; /* Modern font */
     }
-
     .header {
         background-color: #024950;  /* Medium blue header */
         padding: 20px;
@@ -91,7 +95,6 @@ st.markdown("""
         font-weight: bold;
         border-bottom: 1px solid #0FA4AF; /* Lighter blue border */
     }
-
     .chat-box {
         border-radius: 8px;
         padding: 15px;
@@ -101,14 +104,12 @@ st.markdown("""
         background-color: #024950;  /* Medium blue chat box background */
         color: #AFDDE5;             /* Light blue text for chat box */
     }
-
     .user-message {
         text-align: right;
         margin-left: auto;
         background-color: #0FA4AF;  /* Aqua blue for user messages */
         color: #003135;             /* Dark blue text for user messages */
     }
-
     .assistant-message {
         text-align: left;
         margin-right: auto;
@@ -117,7 +118,6 @@ st.markdown("""
         display: flex;
         align-items: center;
     }
-
     .assistant-logo {
         width: 40px;
         height: 40px;
@@ -128,19 +128,16 @@ st.markdown("""
         justify-content: center;
         align-items: center;
     }
-
     .assistant-logo img {
         width: 24px;
         height: 24px;
     }
-
     .message {
         display: inline-block;
         padding: 10px;
         border-radius: 15px;
         width: fit-content;
     }
-
     .stButton button {
         background-color: #0FA4AF;  /* Aqua blue button color */
         color: #003135;            /* Dark blue text on button */
@@ -150,11 +147,9 @@ st.markdown("""
         cursor: pointer;
         font-size: 16px;
     }
-
     .stButton button:hover {
         background-color: #024950; /* Medium blue on hover */
     }
-
     .stTextArea textarea {
         background-color: #024950;  /* Medium blue background for text area */
         color: #AFDDE5;             /* Light blue text for text area */
@@ -164,11 +159,9 @@ st.markdown("""
         width: 100%;
         resize: none;
     }
-
     .stTextArea textarea::placeholder {
         color: #AFDDE5;             /* Light blue placeholder text */
     }
-
     .subheader {
         color: #AFDDE5;             /* Light blue color for subheaders */
         font-size: 20px;
@@ -215,7 +208,7 @@ with st.container():
     # Submit button
     if st.button("Send"):
         if user_input:
-            assistant = OpenAIPAF()
+            assistant = GroqPAF()
             assistant.setup()
 
             # Build prompt and get response
@@ -244,4 +237,4 @@ with st.container():
         else:
             st.warning("Please enter a message.")
 
-st.write("This chatbot uses GPT-4o-mini model to respond to your messages.")
+st.write("This chatbot uses Groq's Mixtral 8x7B model to respond to your messages.")