Update app.py

app.py

@@ -4,6 +4,7 @@ import matplotlib.pyplot as plt
 import zipfile
 import os
 import subprocess
+from datetime import datetime
 
 # Flag to track if subprocess commands have been executed
 subprocess_executed = False
@@ -90,6 +91,18 @@ if uploaded_zip_file is not None:
             # Set the flag to indicate that subprocess commands have been executed
             subprocess_executed = True
 
+            current_datetime = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
+            video_path = f"/home/user/app/C2C/comp2comp/outputs/{current_datetime}/temp_dicom_dir/images/summary/aaa.mp4"
+            image_path = f"/home/user/app/C2C/comp2comp/outputs/{current_datetime}/temp_dicom_dir/images/summary/diameter_graph.png"
+        
+            # Display the video generated by the inference script
+            if os.path.exists(video_path):
+                st.video(video_path, format="video/mp4")
+        
+            # Display the image generated by the inference script
+            if os.path.exists(image_path):
+                st.image(image_path, caption="Diameter Graph", use_column_width=True)
+
     except Exception as e:
         st.error(f"Error: {str(e)}")
     finally: