Update app.py

app.py

@@ -111,6 +111,7 @@ def generate_pdf_report(phase, weather, absentee_pct, delay_log, prediction, ris
         
         # Register DejaVuSans font to support Unicode emojis
         try:
+            # Assuming DejaVuSans.ttf is available in the project directory
             pdfmetrics.registerFont(TTFont('DejaVuSans', 'DejaVuSans.ttf'))
             c.setFont("DejaVuSans", 12)
             flag_indicator = " 🚩" if prediction >= 75 else ""
@@ -298,29 +299,26 @@ try:
         with gr.Row():
             absentee_input = gr.Number(label="Absentee %", value=0)
             delay_input = gr.Number(label="Previous Delay Log", value=0)
-        with gr.Row():
-            output_text = gr.Textbox(label="Prediction Summary")
-            output_pdf = gr.File(label="Download PDF Report")  # Add File component for PDF
+        output = gr.Textbox(label="Prediction Summary")
         submit = gr.Button("Predict")
 
         def predict_and_format(phase, weather, absentee, delay_log):
             print("Gradio predict button clicked.")
             prediction, risk, insight, pdf_base64, pdf_path, heatmap_path = predict_delay(phase, weather, absentee, delay_log)
             if prediction is None:
-                return f"Error: {insight}", None
+                return f"Error: {insight}"
             flag = " 🚩" if prediction >= 75 else ""
-            summary = (f"Predicted Delay: {prediction}%{flag}\n"
-                       f"Risk Level: {risk}\n"
-                       f"Insight: {insight}\n"
-                       f"Heatmap: {'Saved locally at ' + heatmap_path if heatmap_path else 'Failed to generate'}\n"
-                       f"PDF Base64: {'Generated' if pdf_base64 else 'Not generated'}")
-            # Return the summary text and the PDF file path for the File component
-            return summary, pdf_path if pdf_path and os.path.exists(pdf_path) else None
+            return (f"Predicted Delay: {prediction}%{flag}\n"
+                    f"Risk Level: {risk}\n"
+                    f"Insight: {insight}\n"
+                    f"PDF Report: {'Saved locally at ' + pdf_path if pdf_path else 'Failed to generate'}\n"
+                    f"Heatmap: {'Saved locally at ' + heatmap_path if heatmap_path else 'Failed to generate'}\n"
+                    f"PDF Base64: {'Generated' if pdf_base64 else 'Not generated'}")
         
         submit.click(
             predict_and_format,
             inputs=[phase_input, weather_input, absentee_input, delay_input],
-            outputs=[output_text, output_pdf]
+            outputs=output
         )
     print("Gradio UI setup complete.")
 except Exception as e: