Update app.py

app.py

@@ -185,7 +185,7 @@ def generate_report(
         r"\usepackage{amsmath}",
         r"\usepackage{caption}",
         r"\usepackage{pdfpages}",
-        r"\usepackage{times}",  # Using Times font
+        r"\usepackage{times}",
         r"\begin{document}",
         r"\title{NHAI Drone Survey Analysis Report}",
         r"\author{Nagasurendra, Data Analyst}",
@@ -199,7 +199,7 @@ def generate_report(
         r"\item \textbf{Highway Section}: Km 100 to Km 150",
         r"\item \textbf{State}: Telangana",
         r"\item \textbf{Region}: South",
-        rf"\item \textbf{{Survey Date}}: {datetime.now().strftime('%Y-%m-%d')}",  # Fixed line
+        rf"\item \textbf{{Survey Date}}: {datetime.now().strftime('%Y-%m-%d')}",
         r"\item \textbf{Drone Service Provider}: ABC Drone Services Pvt. Ltd.",
         r"\item \textbf{Technology Service Provider}: XYZ AI Analytics Ltd.",
         rf"\item \textbf{{Work Order Reference}}: Data Lake WO-{datetime.now().strftime('%Y-%m-%d')}-XYZ",
@@ -364,16 +364,90 @@ def generate_report(
         # Writing LaTeX file
         with open(tex_path, 'w') as f:
             f.write("\n".join(report_content))
-        log_entries.append(f"LaTeX report saved: {tex_path}")
+        log_entries.append(f"LaTeX file saved: {tex_path}")
 
         # Compiling LaTeX to PDF using latexmk
         import subprocess
-        subprocess.run(["latexmk", "-pdf", "-interaction=nonstopmode", tex_path], cwd=OUTPUT_DIR, check=True)
-        log_entries.append(f"PDF report generated: {report_path}")
+        result = subprocess.run(
+            ["latexmk", "-pdf", "-interaction=nonstopmode", tex_path],
+            cwd=OUTPUT_DIR,
+            capture_output=True,
+            text=True
+        )
+        if result.returncode == 0:
+            log_entries.append(f"PDF report generated: {report_path}")
+            if os.path.exists(report_path):
+                log_entries.append(f"Confirmed report file exists: {report_path}")
+            else:
+                log_entries.append(f"Error: PDF report not found at {report_path}")
+                return ""
+        else:
+            log_entries.append(f"Error: LaTeX compilation failed: {result.stderr}")
+            return ""
         return report_path
     except Exception as e:
         log_entries.append(f"Error: Failed to generate report: {str(e)}")
         return ""
+
+def zip_all_outputs(report_path: str, video_path: str, chart_path: str, map_path: str) -> str:
+    zip_path = os.path.join(OUTPUT_DIR, f"drone_analysis_outputs_{datetime.now().strftime('%Y%m%d_%H%M%S')}.zip")
+    try:
+        with zipfile.ZipFile(zip_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
+            # Add report file
+            if report_path and os.path.exists(report_path):
+                zipf.write(report_path, os.path.basename(report_path))
+                log_entries.append(f"Added report to ZIP: {report_path}")
+            else:
+                log_entries.append(f"Warning: Report file not found for ZIP: {report_path}")
+
+            # Add video file
+            if video_path and os.path.exists(video_path):
+                zipf.write(video_path, os.path.join("outputs", os.path.basename(video_path)))
+                log_entries.append(f"Added video to ZIP: {video_path}")
+            else:
+                log_entries.append(f"Warning: Video file not found for ZIP: {video_path}")
+
+            # Add chart file
+            if chart_path and os.path.exists(chart_path):
+                zipf.write(chart_path, os.path.join("outputs", os.path.basename(chart_path)))
+                log_entries.append(f"Added chart to ZIP: {chart_path}")
+            else:
+                log_entries.append(f"Warning: Chart file not found for ZIP: {chart_path}")
+
+            # Add map file
+            if map_path and os.path.exists(map_path):
+                zipf.write(map_path, os.path.join("outputs", os.path.basename(map_path)))
+                log_entries.append(f"Added map to ZIP: {map_path}")
+            else:
+                log_entries.append(f"Warning: Map file not found for ZIP: {map_path}")
+
+            # Add detected issue images
+            for file in detected_issues:
+                if os.path.exists(file):
+                    zipf.write(file, os.path.join("captured_frames", os.path.basename(file)))
+                    log_entries.append(f"Added image to ZIP: {file}")
+                else:
+                    log_entries.append(f"Warning: Image file not found for ZIP: {file}")
+
+            # Add flight logs
+            for root, _, files in os.walk(FLIGHT_LOG_DIR):
+                for file in files:
+                    file_path = os.path.join(root, file)
+                    if os.path.exists(file_path):
+                        zipf.write(file_path, os.path.join("flight_logs", file))
+                        log_entries.append(f"Added flight log to ZIP: {file_path}")
+                    else:
+                        log_entries.append(f"Warning: Flight log not found for ZIP: {file_path}")
+
+        log_entries.append(f"Created ZIP: {zip_path}")
+        if os.path.exists(zip_path):
+            log_entries.append(f"Confirmed ZIP file exists: {zip_path}")
+        else:
+            log_entries.append(f"Error: ZIP file not created: {zip_path}")
+        return zip_path
+    except Exception as e:
+        log_entries.append(f"Error: Failed to create ZIP: {str(e)}")
+        return ""
 def process_video(video, resize_width=4000, resize_height=3000, frame_skip=5):
     global frame_count, last_metrics, detected_counts, detected_issues, gps_coordinates, log_entries
     frame_count = 0