Update app.py

app.py

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