Update app.py

app.py

@@ -1,4 +1,5 @@
 # Load the packages and configurations
+import os
 import pandas as pd
 from retriever import BM25Plus
 import streamlit as st
@@ -6,7 +7,7 @@ from groq import Groq
 
 # Configure GROQ API_KEY
 client = Groq(
-    api_key="GROQ_API_KEY",
+    api_key=api_key = os.environ.get("GROQ_API_KEY"),
 )
 
 # RTL formatting of the streamlit
@@ -87,18 +88,11 @@ if prompt := st.chat_input():
 
     prompt = get_prompt(prompt, topics)
     response = client.chat.completions.create(
-        #
-        # Required parameters
-        #
         messages=[
-            # Set an optional system message. This sets the behavior of the
-            # assistant and can be used to provide specific instructions for
-            # how it should behave throughout the conversation.
             {
                 "role": "system",
                 "content": "تو یک دستیار سودمند هستی."
             },
-            # Set a user message for the assistant to respond to.
             {
                 "role": "user",
                 "content": prompt,
@@ -107,8 +101,6 @@ if prompt := st.chat_input():
 
         # The language model which will generate the completion.
         model="llama3-70b-8192",
-        #
-        # Optional parameters
     )
     msg = response.choices[0].message.content
     st.session_state.messages.append({"role": "assistant", "content": msg})