Udita feature

app.py

@@ -44,6 +44,14 @@ tax_deductions = '''Extract the following information in the given format:
                 }
 '''
 
+intro = '''Extract the following information in the given format:
+                {
+                'check date': "",
+                'name': "",
+                'address': ""
+                }
+'''
+
 def format_nested_dict(data, indent=0):
     formatted_str = ""
     indent_str = "        " * indent  # Indentation for the current level
@@ -117,13 +125,16 @@ def process_document(image):
     # Process the image with your model
     one = process_function(image_path, other_benifits)
     two = process_function(image_path, tax_deductions)
+    three = process_function(image_path, intro)
+
     text_one = format_nested_dict(one)
     text_two = format_nested_dict(two)
+    text_three = format_nested_dict(three)
 
     # Optionally, you can delete the temporary file after use
     os.remove(image_path)
 
-    return text_one, text_two, one, two
+    return text_one, text_two, text_three, one, two, three
 
 
 
@@ -136,8 +147,11 @@ demo = gr.Interface(
     outputs=[
         gr.Textbox(label="Other Benefits and Information"),  # Second output box with heading
         gr.Textbox(label="Tax Deductions Information"),  # Second output box with heading
+        gr.Textbox(label="Introduction"),  # Second output box with heading
+        
         gr.JSON(label="Other Benefits and Information"),  # First output box with heading
         gr.JSON(label="Tax Deductions Information"),  # First output box with heading
+        gr.JSON(label="Introduction"),
     ],
     title="<div style='text-align: center;'>Information Extraction From PaySlip</div>",
     examples=[["Slip_1.jpg"], ["Slip_2.jpg"]],