Fixed

app.py

@@ -92,7 +92,6 @@ def plot_graph(B, user_risk):
             font_size=10, width=[B[u][v]['weight'] for u, v in B.edges()])
     plt.title("User-Word Bipartite Graph with Risk Heatmap")
     
-    # Capture figure before closing
     fig_canvas = fig
     plt.close(fig)  # release memory
     return fig_canvas
@@ -106,6 +105,10 @@ def analyze_chat(file):
     Processes entirely in memory to avoid storage issues.
     """
     try:
+        # Manual file size check (10 MB max)
+        if file.size > 10_000_000:
+            return "File too large. Max size is 10 MB.", None
+        
         raw_bytes = file.read()
         if file.name.endswith(".csv"):
             df = pd.read_csv(BytesIO(raw_bytes))
@@ -138,7 +141,7 @@ def analyze_chat(file):
 # -------------------------------
 iface = gr.Interface(
     fn=analyze_chat,
-    inputs=gr.File(label="Upload Chat CSV/JSON", file_types=['.csv', '.json'], type='file', max_size=10_000_000),
+    inputs=gr.File(label="Upload Chat CSV/JSON", file_types=['.csv', '.json'], type='file'),
     outputs=[gr.Textbox(label="Metrics"), gr.Plot(label="Risk Graph")],
     title="Sex Chat Detection",
     description="Upload a chat file (CSV or JSON) to detect sex chat, compute metrics, and visualize the user-word graph."