for log, one output simplified

app.py

@@ -43,8 +43,18 @@ tax_deductions = '''Extract the following information in the given format:
                 }
 '''
 
-def demo(image_path, prompt):
-    messages = [
+
+
+
+def process_document(image):
+    # Save the uploaded image to a temporary file
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as tmp_file:
+        image = Image.fromarray(image)  # Convert NumPy array to PIL Image
+        image.save(tmp_file.name)  # Save the image to the temporary file
+        image_path = tmp_file.name  # Get the path of the saved file
+
+
+        messages = [
         {
             "role": "user",
             "content": [
@@ -92,27 +102,11 @@ def demo(image_path, prompt):
         json = literal_eval(almost_json)
     except:
         json = output_text[0]  # Return raw output if JSON parsing fails
-    return json
-
-def process_document(image):
-    # Save the uploaded image to a temporary file
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as tmp_file:
-        image = Image.fromarray(image)  # Convert NumPy array to PIL Image
-        image.save(tmp_file.name)  # Save the image to the temporary file
-        image_path = tmp_file.name  # Get the path of the saved file
-
-    # Process the image with your model
-    one = demo(image_path, other_benifits)
-    two = demo(image_path, tax_deductions)
-    json_op = {
-        "tax_deductions": one,
-        "other_benifits": two
-    }
 
     # Optionally, you can delete the temporary file after use
     os.remove(image_path)
 
-    return json_op
+    return json
 
 # Create Gradio interface
 demo = gr.Interface(