Update app.py

app.py

@@ -0,0 +1,74 @@
+from flask import Flask, request, jsonify, render_template
+from google.cloud import vision
+import io
+import pandas as pd
+import requests
+from salesforce_api import Salesforce
+from datasets import load_dataset
+import os
+
+app = Flask(__name__)
+
+# Initialize Google Cloud Vision client
+vision_client = vision.ImageAnnotatorClient()
+
+# Initialize Salesforce client
+sf = Salesforce(
+    username='your_salesforce_username',
+    password='your_salesforce_password',
+    security_token='your_salesforce_security_token'
+)
+
+# Load dataset from Hugging Face
+dataset = load_dataset('Ajay98/setajsondataset')
+json_data = pd.DataFrame(dataset['train'])
+
+@app.route('/')
+def index():
+    return render_template('index.html')
+
+@app.route('/upload', methods=['POST'])
+def upload_image():
+    if 'file' not in request.files:
+        return "No file part"
+
+    file = request.files['file']
+
+    if file.filename == '':
+        return "No selected file"
+
+    if file:
+        # Extract text from the image
+        extracted_text = extract_text_from_image(file)
+        
+        # Compare extracted text with JSON dataset
+        matched_data = compare_data(extracted_text, json_data)
+
+        # Update Salesforce with structured data
+        update_salesforce(matched_data)
+
+        return jsonify({"extracted_text": extracted_text, "matched_data": matched_data.to_dict(orient='records')})
+
+def extract_text_from_image(image_file):
+    content = image_file.read()
+    image = vision.Image(content=content)
+    response = vision_client.text_detection(image=image)
+    texts = response.text_annotations
+    extracted_text = texts[0].description if texts else ''
+    return extracted_text
+
+def compare_data(extracted_text, json_data):
+    matches = json_data[json_data['column_name'].str.contains(extracted_text, case=False, na=False)]
+    return matches
+
+def update_salesforce(data):
+    for _, row in data.iterrows():
+        sf.Contact.create({
+            'LastName': row['LastName'],
+            'FirstName': row['FirstName'],
+            'Email': row['Email'],
+            'Phone': row['Phone']
+        })
+
+if __name__ == '__main__':
+    app.run(debug=True)