xelpmocAI commited on
Commit
4a7bfda
·
verified ·
1 Parent(s): 9097fe3
Files changed (1) hide show
  1. app.py +13 -3
app.py CHANGED
@@ -98,26 +98,36 @@ def demo(image_path, prompt):
98
  return json
99
 
100
  def process_document(image):
 
 
 
101
  # Save the uploaded image to a temporary file
102
  with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as tmp_file:
103
  image = Image.fromarray(image) # Convert NumPy array to PIL Image
104
  image.save(tmp_file.name) # Save the image to the temporary file
105
  image_path = tmp_file.name # Get the path of the saved file
106
- logging.info("the path made for image: %s", image_path)
 
 
107
  # Process the image with your model
108
  one = demo(image_path, other_benifits)
109
- logging.info("kjf")
 
110
  two = demo(image_path, tax_deductions)
 
 
111
  json_op = {
112
  "tax_deductions": one,
113
  "other_benifits": two
114
  }
115
 
116
- # Optionally, you can delete the temporary file after use
117
  os.remove(image_path)
 
118
 
119
  return json_op
120
 
 
121
  # Create Gradio interface
122
  demo = gr.Interface(
123
  fn=process_document,
 
98
  return json
99
 
100
  def process_document(image):
101
+ # Log the function start
102
+ logging.info("Started processing the document")
103
+
104
  # Save the uploaded image to a temporary file
105
  with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as tmp_file:
106
  image = Image.fromarray(image) # Convert NumPy array to PIL Image
107
  image.save(tmp_file.name) # Save the image to the temporary file
108
  image_path = tmp_file.name # Get the path of the saved file
109
+
110
+ logging.info("The path made for image: %s", image_path)
111
+
112
  # Process the image with your model
113
  one = demo(image_path, other_benifits)
114
+ logging.info("Processed the image for other benefits")
115
+
116
  two = demo(image_path, tax_deductions)
117
+ logging.info("Processed the image for tax deductions")
118
+
119
  json_op = {
120
  "tax_deductions": one,
121
  "other_benifits": two
122
  }
123
 
124
+ # Optionally, delete the temporary file after use
125
  os.remove(image_path)
126
+ logging.info("Temporary file deleted: %s", image_path)
127
 
128
  return json_op
129
 
130
+
131
  # Create Gradio interface
132
  demo = gr.Interface(
133
  fn=process_document,