Create app.py

app.py

@@ -1,36 +1,55 @@
-import os
 import streamlit as st
+import os
 from groq import Groq
 
-# Set the Groq API key (you can also export it as an environment variable)
-groq_api_key = os.getenv('GROQ_API_KEY')  # or replace with your actual API key if not using environment variables
-
-if not groq_api_key:
-    st.error("GROQ_API_KEY is not set. Please set your API key.")
-    st.stop()
-
-# Initialize Groq client
-client = Groq(api_key=groq_api_key)
-
-# Function to get response from Groq API
-def get_groq_response(user_input):
+# Custom function to load variables from a .env file
+def load_dotenv(filepath=".env"):
     try:
-        chat_completion = client.chat.completions.create(
-            messages=[{"role": "user", "content": user_input}],
-            model="llama-3.3-70b-versatile",  # Specify the model you want to use
-        )
-        return chat_completion.choices[0].message.content
+        with open(filepath, "r") as f:
+            for line in f:
+                line = line.strip()
+                # Skip comments and empty lines
+                if not line or line.startswith("#"):
+                    continue
+                if "=" in line:
+                    key, value = line.split("=", 1)
+                    os.environ.setdefault(key.strip(), value.strip())
+    except FileNotFoundError:
+        st.error("No .env file found. Please create one with your API key.")
     except Exception as e:
-        return f"Error occurred: {str(e)}"
+        st.error(f"Error loading .env file: {e}")
 
-# Streamlit app layout
-st.title("Chatbot with Groq API")
+# Load environment variables from the .env file
+load_dotenv()
 
-# Create an input box for the user to type their message
-user_input = st.text_input("Ask me anything:", "")
+def main():
+    st.title("Groq LLM Chatbot")
+    
+    # Fetch API key from environment variables
+    groq_api_key = os.environ.get("GROQ_API_KEY")
+    
+    if groq_api_key:
+        client = Groq(api_key=groq_api_key)
+        st.write("### Chat with LLM")
+        user_input = st.text_area("Enter your message:")
+        
+        if st.button("Generate Response"):
+            if user_input.strip():
+                try:
+                    response = client.chat.completions.create(
+                        model="llama-3.3-70b-versatile",
+                        messages=[
+                            {"role": "user", "content": user_input}
+                        ]
+                    )
+                    st.write("### Response:")
+                    st.write(response.choices[0].message.content)
+                except Exception as e:
+                    st.error(f"Error: {e}")
+            else:
+                st.warning("Please enter a message to generate a response.")
+    else:
+        st.error("API key not found. Please ensure your .env file contains GROQ_API_KEY.")
 
-# If user input is not empty, get the response from Groq
-if user_input:
-    response = get_groq_response(user_input)
-    st.write("### Bot Response:")
-    st.write(response)
+if __name__ == "__main__":
+    main()