4

Dockerfile

@@ -33,4 +33,4 @@ ENV FLASK_APP=app.py
 ENV FLASK_ENV=production
 
 # Run gunicorn
-CMD ["gunicorn", "--bind", "0.0.0.0:7860", "--timeout", "300", "app:app"] 
+CMD ["gunicorn", "--bind", "0.0.0.0:7860", "--timeout", "600", "app:app"] 

app.py

@@ -9,6 +9,7 @@ import tempfile
 from io import BytesIO
 import jinja2
 from dotenv import load_dotenv
+from tenacity import retry, stop_after_attempt, wait_exponential
 
 app = Flask(__name__)
 
@@ -281,6 +282,21 @@ def extract_maintenance_status(report_text):
         logger.error(f"Error extracting maintenance status: {str(e)}")
         return [60, 30, 10]  # Default values if extraction fails
 
+@retry(stop=stop_after_attempt(3), wait=wait_exponential(multiplier=1, min=4, max=10))
+def generate_content_with_retry(prompt, images):
+    """Generate content with retry logic"""
+    try:
+        response = model.generate_content(
+            [prompt] + images,
+            generation_config=genai.types.GenerationConfig(
+                timeout=300
+            )
+        )
+        return response
+    except Exception as e:
+        logger.error(f"Error generating content: {str(e)}")
+        raise
+
 @app.route('/generate_report', methods=['POST'])
 def generate_report():
     try:
@@ -334,7 +350,10 @@ def generate_report():
         contextualized_prompt = contextualized_prompt.replace('${gate_count}', str(basic_info.get('gateCount', '')))
         
         # Generate report with image references
-        response = model.generate_content([contextualized_prompt] + all_images)
+        response = generate_content_with_retry(
+            contextualized_prompt,
+            all_images
+        )
         report_text = response.text
         
         # Extract all metrics
@@ -467,5 +486,19 @@ def health_check():
             'error': str(e)
         }), 500
 
+@app.route('/report_status/<task_id>')
+def report_status(task_id):
+    """Check the status of a report generation task"""
+    try:
+        # Implement status checking logic
+        return jsonify({
+            'status': 'processing',
+            'progress': 50,  # Example progress percentage
+            'message': 'Processing images...'
+        })
+    except Exception as e:
+        logger.error(f"Error checking status: {str(e)}")
+        return jsonify({'error': str(e)}), 500
+
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)