Update app.py

app.py

@@ -136,23 +136,45 @@ def process_traffic_image(image):
     """
     Orchestrates the traffic analysis workflow.
     """
-    try:
-        # Save the uploaded image temp
-        image_path = "temp_traffic_image.jpg"
-        image.save(image_path)
+    # Save the uploaded image temporarily
+    image_path = "temp_traffic_image.jpg"
+    image.save(image_path)
+
+    # Analyze traffic and optimize signal timing
+    with open(image_path, "rb") as img_file:
+        response = requests.post(
+            f"{HOSTED_API_URL}/analyze_traffic/",
+            files={"file": img_file}
+        )
+    
+    os.remove(image_path)  # Clean up the temporary image
+
+    if response.status_code == 200:
+        data = response.json()
+        vehicle_count = data.get("vehicle_count", 0)
+        congestion_level = data.get("congestion_level", "Unknown")
+        flow_rate = data.get("flow_rate", "Unknown")
+        processed_image_base64 = data.get("processed_image", None)
+
+        # Decode the processed image
+        if processed_image_base64:
+            processed_image = Image.open(io.BytesIO(base64.b64decode(processed_image_base64)))
+        else:
+            processed_image = None
 
-        vehicle_count, congestion_level, flow_rate, processed_image = analyze_traffic(image_path)
+        # Signal timing optimization
         signal_timing = optimize_signal_rl(congestion_level)
 
-        os.remove(image_path)  # Clean up
-
-        return (f"Detected Vehicles: {vehicle_count}\n"
-                f"Congestion Level: {congestion_level}\n"
-                f"Traffic Flow: {flow_rate}\n"
-                f"Signal Timing Suggestion: {signal_timing}", processed_image)
-    except Exception as e:
-        logging.error(f"Error processing traffic image: {e}")
-        return "An error occurred while processing the image.", None
+        # Return the results
+        return (
+            f"Detected Vehicles: {vehicle_count}\n"
+            f"Congestion Level: {congestion_level}\n"
+            f"Traffic Flow: {flow_rate}\n"
+            f"Signal Timing Suggestion: {signal_timing}",
+            processed_image
+        )
+    else:
+        return "Error in analyzing traffic.", None
 
 # Gradio Interface
 def gradio_interface(image):