Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 import cv2
 import easyocr
 from simple_salesforce import Salesforce
+import re
 
 # Salesforce credentials
 Salesforce_User_Name = '[email protected]'  # Your Salesforce username
@@ -29,32 +30,45 @@ def extract_patient_info(image):
     result = reader.readtext(image_rgb)
     extracted_text = " ".join([detection[1] for detection in result])
 
-    # Tokenize the extracted text with LayoutLM
-    inputs = tokenizer(extracted_text, return_tensors="pt")
-    outputs = model(**inputs)
-
-    # Here, extracted_text is already available from EasyOCR, we can extract relevant details
+    # Extract relevant details (Name, Age, Gender, Phone number) from the extracted text
     details = extract_details_from_text(extracted_text)
     
     # Create a record in Salesforce using the extracted details
     create_salesforce_record(details)
 
-    # Return the extracted text
+    # Return the extracted text for display
     return extracted_text
 
-# Function to extract details from the extracted text (use regex or other methods to extract)
+# Function to extract details from the extracted text using regex
 def extract_details_from_text(extracted_text):
-    # Simple example of extracting details, customize this according to the format of the text
+    # Regex patterns to match Name, Age, Gender, and Phone number
     details = {}
-    details['Name'] = "Shanthi"  # Here, add the logic to extract the actual name
-    details['Age'] = "39"        # Similarly, extract age, gender, and phone number
-    details['Gender'] = "Female"
-    details['Phone Number'] = "9955337097"
-    
+
+    # Extract Name
+    name_match = re.search(r"Name[:\s]*([A-Za-z\s]+)", extracted_text)
+    if name_match:
+        details['Name'] = name_match.group(1)
+
+    # Extract Age
+    age_match = re.search(r"Age[:\s]*([\d]+)", extracted_text)
+    if age_match:
+        details['Age'] = age_match.group(1)
+
+    # Extract Gender
+    gender_match = re.search(r"Gender[:\s]*(Male|Female)", extracted_text, re.IGNORECASE)
+    if gender_match:
+        details['Gender'] = gender_match.group(1)
+
+    # Extract Phone number
+    phone_match = re.search(r"Phone number[:\s]*([\d]+)", extracted_text)
+    if phone_match:
+        details['Phone Number'] = phone_match.group(1)
+
     return details
 
-# Function to create a record in Salesforce
+# Function to create a record in Salesforce using the extracted details
 def create_salesforce_record(details):
+    # Prepare the data to be inserted into Salesforce
     data = {
         'Name__c': details['Name'],
         'Age__c': int(details['Age']),