Update app.py

app.py

@@ -10,6 +10,7 @@ load_dotenv()
 
 app = Flask(__name__, template_folder='templates', static_folder='static')
 
+# Salesforce Connection
 def get_salesforce_connection():
     try:
         sf = Salesforce(
@@ -23,6 +24,7 @@ def get_salesforce_connection():
         print(f"Error connecting to Salesforce: {e}")
         return None
 
+# Initialize Salesforce connection
 sf = get_salesforce_connection()
 
 @app.route('/')
@@ -33,85 +35,39 @@ def index():
 def serve_static(filename):
     return send_from_directory('static', filename)
 
-@app.route('/get_ingredients', methods=['POST'])
-def get_ingredients():
-    dietary_preference = request.json.get('dietary_preference', '').strip().lower()
-    logging.debug(f"Received dietary preference: {dietary_preference}")
-
-    # Map dietary preference to SOQL condition
-    if dietary_preference == 'both':
-        condition = "Category__c = 'both'"  # This will fetch both vegetarian and non-vegetarian
-    else:
-        preference_map = {
-            'vegetarian': "Category__c = 'Veg'",
-            'non-vegetarian': "Category__c = 'Non-Veg'"
-        }
-        condition = preference_map.get(dietary_preference)
-
-    if not condition:
-        logging.debug("Invalid dietary preference received.")
-        return jsonify({"error": "Invalid dietary preference."}), 400
-
+# Sample endpoint for Salesforce queries (non-restaurant)
+@app.route('/get_salesforce_data', methods=['GET'])
+def get_salesforce_data():
     try:
-        soql = f"SELECT Name, Image_URL__c FROM Sector_Detail__c WHERE {condition} LIMIT 200"
+        # Example SOQL query for retrieving data from a Salesforce object
+        soql = "SELECT Id, Name FROM Account LIMIT 10"
         result = sf.query(soql)
-        ingredients = [
-            {"name": record['Name'], "image_url": record.get('Image_URL__c', '')}
-            for record in result['records'] if 'Name' in record
-        ]
-        logging.debug(f"Fetched {len(ingredients)} ingredients.")
-        return jsonify({"ingredients": ingredients})
-    except Exception as e:
-        logging.error(f"Error while fetching ingredients: {str(e)}")
-        return jsonify({"error": f"Failed to fetch ingredients: {str(e)}"}), 500
-
-
-@app.route('/get_menu_items', methods=['POST'])
-def get_menu_items():
-    ingredient_names = request.json.get('ingredient_names', '').strip().lower()
-    logging.debug(f"Received ingredient names: {ingredient_names}")
-
-    # Constructing a SOQL query where Name contains any of the ingredient names
-    # Split the ingredient names into separate words to match each ingredient
-    ingredients_list = ingredient_names.split()
-
-    # Create a dynamic WHERE clause to match the ingredient names in the Menu Item names
-    condition = " OR ".join([f"Name LIKE '%{ingredient}%'" for ingredient in ingredients_list])
-
-    if not condition:
-        logging.debug("No ingredients received.")
-        return jsonify({"error": "No valid ingredients provided."}), 400
 
-    try:
-        soql = f"SELECT Name, Price__c, Description__c, Image1__c, Image2__c, Veg_NonVeg__c, Section__c, Total_Ordered__c FROM Menu_Item__c WHERE {condition} LIMIT 200"
-        result = sf.query(soql)
-        
-        menu_items = [
-            {"name": record['Name'], "price": record.get('Price__c', ''), "description": record.get('Description__c', ''), 
-             "image_url1": record.get('Image1__c', ''), "image_url2": record.get('Image2__c', ''), 
-             "veg_nonveg": record.get('Veg_NonVeg__c', ''), "section": record.get('Section__c', ''), 
-             "total_ordered": record.get('Total_Ordered__c', '')}
-            for record in result['records'] if 'Name' in record
-        ]
-        logging.debug(f"Fetched {len(menu_items)} menu items based on ingredients.")
-        return jsonify({"menu_items": menu_items})
-    
+        # Format the result as needed
+        records = [{"id": record['Id'], "name": record['Name']} for record in result['records']]
+        logging.debug(f"Fetched {len(records)} records from Salesforce.")
+        return jsonify({"data": records})
     except Exception as e:
-        logging.error(f"Error while fetching menu items: {str(e)}")
-        return jsonify({"error": f"Failed to fetch menu items: {str(e)}"}), 500
-
-
+        logging.error(f"Error while fetching data from Salesforce: {str(e)}")
+        return jsonify({"error": f"Failed to fetch data: {str(e)}"}), 500
 
-@app.route('/submit_ingredients', methods=['POST'])
-def submit_ingredients():
+# Example of a POST endpoint for submitting data (non-restaurant)
+@app.route('/submit_salesforce_data', methods=['POST'])
+def submit_salesforce_data():
     data = request.json
-    ingredients = data.get('ingredients', [])
+    record_data = data.get('record', {})
     
-    if not ingredients:
-        return jsonify({'error': 'No ingredients selected'}), 400
-    
-    logging.debug(f"Ingredients submitted: {ingredients}")
-    return jsonify({'success': True})
+    if not record_data:
+        return jsonify({'error': 'No data provided'}), 400
+
+    try:
+        # Example of creating a new Salesforce record
+        account = sf.Account.create(record_data)
+        logging.debug(f"Created new Salesforce account with Id: {account['id']}")
+        return jsonify({'success': True, 'account_id': account['id']})
+    except Exception as e:
+        logging.error(f"Error while submitting data to Salesforce: {str(e)}")
+        return jsonify({"error": f"Failed to submit data: {str(e)}"}), 500
 
 if __name__ == '__main__':
-    app.run(debug=True, host='0.0.0.0', port=7860) 
+    app.run(debug=True, host='0.0.0.0', port=7860)