Update app.py

app.py

@@ -9,6 +9,7 @@ import logging
 import cv2
 import numpy as np
 import traceback
+from datetime import datetime
 
 # Set up logging
 logging.basicConfig(level=logging.DEBUG)
@@ -193,6 +194,31 @@ def chat_about_image(message, chat_history):
         logger.error(f"Error during chat: {str(e)}")
         return "", chat_history + [(message, f"Error: {str(e)}")]
 
+def generate_summary_report(chat_history):
+    """
+    Generate a summary report from the chat history.
+    """
+    report = "Construction Site Safety Analysis Report\n"
+    report += "=" * 40 + "\n"
+    report += f"Generated on: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}\n\n"
+
+    for i, (user, ai) in enumerate(chat_history, 1):
+        if user:
+            report += f"Query {i}:\n{user}\n\n"
+        if ai:
+            report += f"Analysis {i}:\n{ai}\n\n"
+        report += "-" * 40 + "\n"
+
+    return report
+
+def download_report(chat_history):
+    """
+    Generate and provide a download link for the summary report.
+    """
+    report = generate_summary_report(chat_history)
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    filename = f"safety_analysis_report_{timestamp}.txt"
+    return (filename, report)
 
 # Custom CSS for improved styling
 custom_css = """
@@ -258,9 +284,14 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as iface:
             elem_classes="chat-input"
         )
 
-    # Fifth row: Clear Chat Button
+    # Fifth row: Clear Chat and Download Report Buttons
     with gr.Row():
-        clear = gr.Button("🗑️ Clear", elem_classes="clear-button")
+        clear = gr.Button("🗑️ Clear Chat", elem_classes="clear-button")
+        download_button = gr.Button("📥 Download Report", elem_classes="download-button")
+
+    # Add a file component to handle the download
+    report_file = gr.File(label="Download Safety Analysis Report")
+
 
     def update_chat(history, new_messages):
         history = history or []
@@ -278,25 +309,24 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as iface:
     msg.submit(chat_about_image, [msg, chatbot], [msg, chatbot])
     clear.click(lambda: None, None, chatbot, queue=False)
 
+
+  download_button.click(
+        download_report,
+        inputs=[chatbot],
+        outputs=[report_file]
+    )
+
     gr.HTML(
         """
         <div class="groq-badge">Powered by Groq</div>
         """
     )
 
+# Launch the app
 if __name__ == "__main__":
     try:
-        # Attempt to launch with sharing enabled
-        iface.launch(debug=True, share=True)
+        iface.launch(debug=True)
     except Exception as e:
-        logger.error(f"Error when trying to launch with sharing: {str(e)}")
+        logger.error(f"Error when trying to launch the interface: {str(e)}")
         logger.error(traceback.format_exc())
-        
-        # If sharing fails, try to launch without sharing
-        try:
-            logger.info("Attempting to launch without sharing...")
-            iface.launch(debug=True, share=False)
-        except Exception as e:
-            logger.error(f"Error when trying to launch without sharing: {str(e)}")
-            logger.error(traceback.format_exc())
-            print("Failed to launch the Gradio interface. Please check the logs for more information.")
+        print("Failed to launch the Gradio interface. Please check the logs for more information.")