Update app.py

app.py

@@ -1,19 +1,27 @@
+import os
 import streamlit as st
 import easyocr
 import openai
-from openai import OpenAI
+from dotenv import load_dotenv
 from PIL import Image
-import numpy as np
 import io
 
-# ✅ Fix: Ensure set_page_config is the first Streamlit command
-st.set_page_config(page_title="Multimodal AI Assistant", layout="wide")
+# ✅ Load API key from .env
+load_dotenv()
+api_key = os.getenv("GROQ_API_KEY")
+
+if not api_key:
+    st.error("❌ API key not found! Please set `GROQ_API_KEY` in your `.env` file.")
+    st.stop()
 
-# ✅ OpenAI API Key Setup (Replace with your actual key)
-client = OpenAI(api_key="your_openai_api_key")  # Set your OpenAI API key
+# ✅ Set up OpenAI (Groq API)
+openai.api_key = api_key
+
+# ✅ Ensure Streamlit config is first
+st.set_page_config(page_title="Multimodal AI Assistant", layout="wide")
 
 # ✅ Initialize OCR Reader
-reader = easyocr.Reader(['en'])
+reader = easyocr.Reader(["en"])
 
 # ✅ Streamlit App Layout
 st.title("📸 Multimodal AI Assistant")
@@ -27,28 +35,38 @@ if uploaded_file:
     image = Image.open(uploaded_file)
     st.image(image, caption="Uploaded Image", use_container_width=True)
 
-    # ✅ Convert image to a format that EasyOCR supports
+    # ✅ Convert image to bytes for EasyOCR
     image_bytes = io.BytesIO(uploaded_file.getvalue()).read()
-    extracted_text = reader.readtext(image_bytes, detail=0)
+
+    # ✅ Extract text using EasyOCR
+    with st.spinner("🔍 Extracting text..."):
+        extracted_text = reader.readtext(image_bytes, detail=0)
 
     # ✅ Show extracted text
     extracted_text_str = " ".join(extracted_text) if extracted_text else "No text found"
     st.subheader("📝 Extracted Text:")
     st.write(extracted_text_str)
 
-    # ✅ Ask a question about the extracted text
+    # ✅ Question Answering Section
     user_query = st.text_input("Ask a question about the extracted text:")
 
-    if user_query:
-        with st.spinner("Thinking... 💭"):
-            response = client.chat.completions.create(
-                model="gpt-3.5-turbo",
-                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
-            st.subheader("🤖 AI Answer:")
-            st.write(answer)
+    if st.button("Get Answer"):
+        if not extracted_text_str or extracted_text_str == "No text found":
+            st.warning("⚠ No text available for analysis.")
+        elif user_query.strip() == "":
+            st.warning("⚠ Please enter a question.")
+        else:
+            with st.spinner("🤖 Thinking..."):
+                try:
+                    response = openai.ChatCompletion.create(
+                        model="llama3-70b-8192",  # Update model if needed
+                        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"]
+                    st.subheader("🤖 AI Answer:")
+                    st.write(answer)
+                except openai.error.OpenAIError as e:
+                    st.error(f"❌ API Error: {e}")