2

.gitignore

@@ -3,4 +3,6 @@ __pycache__/
 .env
 .venv
 venv/
-ENV/ 
+ENV/
+*.log
+.DS_Store 

app.py

@@ -12,15 +12,25 @@ from dotenv import load_dotenv
 
 app = Flask(__name__)
 
+# Load environment variables
+load_dotenv()
+
 # Configure logging
 logging.basicConfig(level=logging.DEBUG)
 logger = logging.getLogger(__name__)
 
-# Configure Gemini API
-load_dotenv()
-genai.configure(api_key=os.getenv('GOOGLE_API_KEY'))
-# Use gemini-pro-vision model for image processing
-model = genai.GenerativeModel("gemini-1.5-flash")
+# Configure Gemini API with error handling
+api_key = os.getenv('GOOGLE_API_KEY')
+if not api_key:
+    logger.error("No Google API key found. Please set GOOGLE_API_KEY environment variable.")
+    raise ValueError("GOOGLE_API_KEY environment variable is required")
+
+try:
+    genai.configure(api_key=api_key)
+    model = genai.GenerativeModel("gemini-1.5-flash")
+except Exception as e:
+    logger.error(f"Failed to configure Gemini API: {str(e)}")
+    raise
 
 # Updated prompt for dual-format report
 prompt = """You are a professional campus facility inspector with over 15 years of experience in infrastructure assessment in India. Analyze the campus with total area of ${college_area} acres. Generate two reports based on the provided campus images:
@@ -268,6 +278,9 @@ def extract_maintenance_status(report_text):
 @app.route('/generate_report', methods=['POST'])
 def generate_report():
     try:
+        if not api_key:
+            raise ValueError("Google API key not configured")
+
         data = request.json
         images = data.get('images', [])
         basic_info = data.get('basicInfo', {})
@@ -352,9 +365,12 @@ def generate_report():
             } for img in image_contexts]
         })
         
+    except ValueError as ve:
+        logger.error(f"Validation error: {str(ve)}")
+        return jsonify({'error': str(ve)}), 400
     except Exception as e:
         logger.error(f"Error generating report: {str(e)}")
-        return jsonify({'error': str(e)}), 500
+        return jsonify({'error': 'Internal server error occurred'}), 500
 
 @app.route('/download_pdf', methods=['POST'])
 def download_pdf():
@@ -429,5 +445,21 @@ def generate_report(findings, inspector_name, location, weather):
     
     return html_content
 
+@app.route('/health')
+def health_check():
+    try:
+        # Simple test to verify API connection
+        response = model.generate_content("Test connection")
+        return jsonify({
+            'status': 'healthy',
+            'api_configured': True
+        })
+    except Exception as e:
+        logger.error(f"Health check failed: {str(e)}")
+        return jsonify({
+            'status': 'unhealthy',
+            'error': str(e)
+        }), 500
+
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)