Update app.py

app.py

@@ -49,22 +49,27 @@ def yolo_and_trocr(image_input, save):
     # Saving the info in a dictionary
     if save: 
       data_dict = ast.literal_eval(decoded_text)
-      with open(f"{data_dict['Address']}.json", "w") as file:
+      file_path = f"{data_dict['Address']}.json"
+      with open(file_path, "w") as file:
         current_datetime = datetime.now()
         timestamp = current_datetime.strftime("%Y-%m-%d %H:%M:%S")
         data_dict['Last_Reading'] = {f'{timestamp}': f'{text}'}
-        json.dump(data_dict, file, indent=4)
+        return image, text, decoded_text, file_path 
 
-    
-    return image, text, decoded_text
+    else:
+      return image, text, decoded_text, None
   
   except Exception as e:
     return "", f"Your input is invalid: {str(e)}", f"Try Again: Make sure the meter and QR code are clearly captured"
 
 app = gr.Interface(
     fn=yolo_and_trocr,
-    inputs=[gr.File(label="Input: Water Meter Image"), gr.Checkbox(label="Save")], 
-    outputs=[gr.Image(label='Output: Water Meter Photo'), gr.Textbox(label="Output: Water Meter Reading"), gr.Textbox(label="Output: QR Code Detection")],
+    inputs=[gr.File(label="Input: Water Meter Image"),
+            gr.Checkbox(label="Save")], 
+    outputs=[gr.Image(label='Output: Water Meter Photo'),
+             gr.Textbox(label="Output: Water Meter Reading"),
+             gr.Textbox(label="Output: QR Code Detection"),
+             gr.File(label="Output: Saved Data")],
     title="Water Meter Reading with YOLO and OCR"
 )