Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,5 +1,4 @@
 import streamlit as st
-#import openai 
 from dotenv import load_dotenv
 import os
 from groq import Groq
@@ -21,8 +20,16 @@ st.markdown("Upload an image or provide an image URL to get AI-generated comment
 # Load environment variables
 load_dotenv()
 
-# Initialize Groq client
-client = Groq(api_key=os.getenv("GROQ_API_KEY"))
+# Initialize Groq client with the correct configuration
+api_key = os.getenv("GROQ_API_KEY")
+if not api_key:
+    st.error("Please set your GROQ_API_KEY in the .env file")
+    st.stop()
+
+client = Groq(
+    api_key=api_key,
+    base_url="https://api.groq.com/v1"
+)
 
 # Create input field for image URL
 image_url = st.text_input("Enter Image URL", "https://static.seekingalpha.com/uploads/2016/1/19/saupload_fredgraph.jpg")
@@ -49,7 +56,7 @@ if st.button("Analyze Image"):
             
             # Create the completion request
             completion = client.chat.completions.create(
-                model="meta-llama/llama-4-scout-17b-16e-instruct",
+                model="llama2-70b-4096",  # Updated model name
                 messages=[
                     {
                         "role": "user",
@@ -68,10 +75,9 @@ if st.button("Analyze Image"):
                     }
                 ],
                 temperature=1,
-                max_completion_tokens=300,
+                max_tokens=300,  # Updated parameter name
                 top_p=1,
-                stream=False,
-                stop=None,
+                stream=False
             )
 
             # Display the image