Update app.py

app.py

@@ -4,6 +4,9 @@ from PIL import Image
 import easyocr
 import os
 from groq import Groq
+from dotenv import load_dotenv
+
+load_dotenv() # Load environment variables from .env
 
 # OCR Function
 def extract_text_from_image(image):
@@ -34,8 +37,12 @@ def groq_chat(prompt):
             model="llama-3.3-70b-versatile",
         )
         return chat_completion.choices[0].message.content
+    except groq.APIConnectionError as e:
+        return f"Groq API Connection Error: {e}"
+    except groq.AuthenticationError as e:
+        return f"Groq API Authentication Error: {e}"
     except Exception as e:
-        return f"Error using Groq API: {e}. Please ensure GROQ_API_KEY is set correctly."
+        return f"General Groq API Error: {e}"
 
 # Streamlit App
 def main():
@@ -45,7 +52,7 @@ def main():
 
     if uploaded_file is not None:
         image = Image.open(uploaded_file)
-        st.image(image, caption="Uploaded Image", use_column_width=True)
+        st.image(image, caption="Uploaded Image", use_container_width=True) # Changed here
 
         if st.button("Extract Text and Enable Question Answering"):
             with st.spinner("Extracting text..."):