Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 import streamlit as st
 import easyocr
-import openai
+from openai import OpenAI, OpenAIError  # ✅ Correct import for API errors
 from dotenv import load_dotenv
 from PIL import Image
 import io
@@ -14,8 +14,8 @@ if not api_key:
     st.error("❌ API key not found! Please set `GROQ_API_KEY` in your `.env` file.")
     st.stop()
 
-# ✅ Set up OpenAI (Groq API)
-openai.api_key = api_key
+# ✅ Initialize OpenAI client for Groq API
+client = OpenAI(api_key=api_key)
 
 # ✅ Ensure Streamlit config is first
 st.set_page_config(page_title="Multimodal AI Assistant", layout="wide")
@@ -58,15 +58,15 @@ if uploaded_file:
         else:
             with st.spinner("🤖 Thinking..."):
                 try:
-                    response = openai.ChatCompletion.create(
-                        model="llama3-70b-8192",  # Update model if needed
+                    response = client.chat.completions.create(  # ✅ Corrected API call
+                        model="llama3-70b-8192",  # ✅ Groq LLaMA 3 model
                         messages=[
                             {"role": "system", "content": "You are an AI assistant analyzing extracted text from images."},
                             {"role": "user", "content": f"Extracted text: {extracted_text_str}\n\nUser question: {user_query}"}
                         ]
                     )
-                    answer = response["choices"][0]["message"]["content"]
+                    answer = response.choices[0].message.content
                     st.subheader("🤖 AI Answer:")
                     st.write(answer)
-                except openai.error.OpenAIError as e:
+                except OpenAIError as e:
                     st.error(f"❌ API Error: {e}")