Upload app.py

app.py

@@ -31,11 +31,11 @@ def analyze_pcap(pcap_file):
 
     # Prepare outputs
     html_report_path = os.path.join(output_dir, "sip_rtp_report.html")
-    rtp_graphs = [f for f in os.listdir(output_dir) if f.startswith("rtp_visualization") and f.endswith(".png")]
-    sequence_diagrams = [f for f in os.listdir(output_dir) if f.startswith("sip_sequence") and f.endswith(".png")]
+    rtp_graphs = [os.path.join(output_dir, f) for f in os.listdir(output_dir) if f.startswith("rtp_visualization") and f.endswith(".png")]
+    sequence_diagrams = [os.path.join(output_dir, f) for f in os.listdir(output_dir) if f.startswith("sip_sequence") and f.endswith(".png")]
     dashboard = os.path.join(output_dir, "call_quality_dashboard.png")
 
-    return result_text, html_report_path, [os.path.join(output_dir, f) for f in sequence_diagrams], [os.path.join(output_dir, f) for f in rtp_graphs], dashboard
+    return result_text, html_report_path, sequence_diagrams, rtp_graphs, dashboard
 
 with gr.Blocks(title="SIP & RTP PCAP Analyzer") as demo:
     gr.Markdown("""
@@ -48,8 +48,8 @@ with gr.Blocks(title="SIP & RTP PCAP Analyzer") as demo:
 
     result_text = gr.Textbox(label="Analysis Summary", lines=15)
     html_report = gr.File(label="Download HTML Report")
-    sequence_images = gr.Gallery(label="SIP Sequence Diagrams").style(grid=2)
-    rtp_images = gr.Gallery(label="RTP Stream Visualizations").style(grid=2)
+    sequence_images = gr.Gallery(label="SIP Sequence Diagrams")
+    rtp_images = gr.Gallery(label="RTP Stream Visualizations")
     dashboard_image = gr.Image(label="Call Quality Dashboard")
 
     run_btn.click(