Update app.py

app.py

@@ -16,6 +16,10 @@ from reportlab.lib.pagesizes import letter
 from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Table, TableStyle
 from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
 import io
+from dotenv import load_dotenv
+
+# Load environment variables
+load_dotenv()
 
 # Configure logging
 logging.basicConfig(
@@ -26,8 +30,8 @@ logger = logging.getLogger(__name__)
 
 # Configuration and Constants
 class Config:
-    GEMINI_URL = "https://generativelanguage.googleapis.com/v1beta/models/gemini-pro-vision:generateContent"
-    GEMINI_API_KEY = os.getenv("GEMINI_API_KEY", "YOUR_API_KEY")
+    GEMINI_URL = "https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent"
+    GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
     MAX_RETRIES = 3
     TIMEOUT = 30
     MAX_IMAGE_SIZE = (1600, 1600)
@@ -175,6 +179,7 @@ class PDFGenerator:
         doc.build(elements)
         return buffer.getvalue()
 
+
 class ImageProcessor:
     @staticmethod
     def validate_image(uploaded_file) -> tuple[bool, str]: