Update app.py

app.py

@@ -6,17 +6,15 @@ import time
 
 # Constants for file paths and MIME types
 PDF_MIME_TYPE = "application/pdf"
-
-# ✅ Store temporary files in /tmp/ (Hugging Face safe storage)
-PRODUCT_IMAGES_DIR = "/tmp/product"
-COMPETITOR_IMAGES_DIR = "/tmp/competitor"
-REPORT = "/tmp/report.pdf"
-STATUS_FILE = "/tmp/status.txt"
-BRAND_MARKETING = "/tmp/brand_marketing.pdf"
-CONTENT_MARKETING = "/tmp/content_marketing.pdf"
-SOCIAL_MEDIA_MARKETING = "/tmp/social_media_marketing.pdf"
-
-# Ensure directories exist
+BRAND_MARKETING = "data/reports/template_PDF/brand marketing.pdf"
+CONTENT_MARKETING = "data/reports/template_PDF/content marketing.pdf"
+SOCIAL_MEDIA_MARKETING = "data/reports/template_PDF/social media marketing.pdf"
+PRODUCT_IMAGES_DIR = "data/product"
+COMPETITOR_IMAGES_DIR = "data/competitor"
+REPORT = "src/Report/report.pdf"
+STATUS_FILE = "src/Report/status.txt"  # Path to the status file
+
+# Ensure directories for saving images exist
 os.makedirs(PRODUCT_IMAGES_DIR, exist_ok=True)
 os.makedirs(COMPETITOR_IMAGES_DIR, exist_ok=True)
 
@@ -92,7 +90,7 @@ if st.button("Generate"):
                 st.session_state.analysis_in_progress = True  # mark as in progress
                 st.subheader("Product Images vs Competitor Images")
 
-                # ✅ Save images to /tmp/ (cloud-safe location)
+                # Save images to respective directories
                 for idx, img_file in enumerate(product_images):
                     img = Image.open(img_file)
                     if img.mode == "RGBA":
@@ -110,8 +108,8 @@ if st.button("Generate"):
                     with open(STATUS_FILE, "w") as f:
                         f.write("pending")
 
-                    # ✅ Execute analysis script with /tmp/ directory
-                    subprocess.Popen(['python', 'src/analysis.py', company_name, "/tmp/"])
+                    # Execute the analysis script asynchronously
+                    subprocess.Popen(['python', 'src/analysis.py', company_name])
 
                     # Monitor the status file
                     st.info("Running analysis. Please wait...")
@@ -121,7 +119,7 @@ if st.button("Generate"):
                         st.success("Analysis completed successfully!")
                         st.session_state.report_generated = True  # mark as generated
 
-                        # ✅ Provide Download Buttons for Generated PDF Reports
+                        # Download Buttons for Generated PDF Reports
                         for report_name, label in [
                             (BRAND_MARKETING, f"{company_name} Brand Marketing Report"),
                             (CONTENT_MARKETING, f"{company_name} Content Marketing Report"),
@@ -138,8 +136,8 @@ if st.button("Generate"):
                             else:
                                 st.error(f"{report_name} not found. Please generate the report first.")
                         
-                        # ✅ Provide Download for the Final Report
-                        custom_report_name = f"/tmp/{company_name}_report.pdf"
+                        # For the custom report
+                        custom_report_name = f"{company_name} report.pdf"
                         if os.path.exists(REPORT):
                             with open(REPORT, "rb") as report_file:
                                 st.download_button(