Spaces:
Sleeping
Sleeping
| from flask import Flask, render_template, send_from_directory, request, jsonify | |
| from simple_salesforce import Salesforce | |
| from dotenv import load_dotenv | |
| import os | |
| import openai | |
| openai.api_key = "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" | |
| # Load environment variables from .env file | |
| load_dotenv() | |
| app = Flask(__name__, template_folder='templates', static_folder='static') | |
| # Function to get Salesforce connection | |
| def get_salesforce_connection(): | |
| try: | |
| sf = Salesforce( | |
| username=os.getenv('SFDC_USERNAME'), | |
| password=os.getenv('SFDC_PASSWORD'), | |
| security_token=os.getenv('SFDC_SECURITY_TOKEN'), | |
| domain=os.getenv('SFDC_DOMAIN', 'login') | |
| ) | |
| return sf | |
| except Exception as e: | |
| print(f"Error connecting to Salesforce: {e}") | |
| return None | |
| # Initialize Salesforce connection | |
| sf = get_salesforce_connection() | |
| def index(): | |
| return render_template('index.html') | |
| def serve_static(filename): | |
| return send_from_directory('static', filename) | |
| def get_ingredients(): | |
| dietary_preference = request.json.get('dietary_preference', '').strip().lower() | |
| # Debugging: print the received dietary preference | |
| print(f"Received dietary preference: {dietary_preference}") | |
| if dietary_preference == 'veg': | |
| print("Fetching ingredients for Vegetarian...") # Debug when fetching vegetarian ingredients | |
| soql = "SELECT Name, Image_URL__c FROM Sector_Detail__c WHERE Category__c IN ('Veg') LIMIT 200" | |
| elif dietary_preference == 'non-vegetarian': | |
| print("Fetching ingredients for Non-Vegetarian...") # Debug when fetching non-vegetarian ingredients | |
| soql = "SELECT Name, Image_URL__c FROM Sector_Detail__c WHERE Category__c IN ('Non-Veg') LIMIT 200" | |
| else: | |
| print("Invalid dietary preference received.") # Debug for invalid dietary preference | |
| return jsonify({"error": "Invalid dietary preference."}), 400 | |
| try: | |
| result = sf.query(soql) | |
| ingredients = [ | |
| {"name": record['Name'], "image_url": record.get('Image_URL__c', '')} | |
| for record in result['records'] if 'Name' in record | |
| ] | |
| print(f"Fetched {len(ingredients)} ingredients.") # Debug: print how many ingredients were fetched | |
| return jsonify({"ingredients": ingredients}) | |
| except Exception as e: | |
| print(f"Error while fetching ingredients: {str(e)}") # Debug error | |
| return jsonify({"error": f"Failed to fetch ingredients: {str(e)}"}), 500 | |
| def submit_ingredients(): | |
| data = request.json | |
| ingredients = data.get('ingredients', []) | |
| if not ingredients: | |
| return jsonify({'error': 'No ingredients selected'}), 400 | |
| # You can now process the selected ingredients, store them in a database, etc. | |
| print(f"Ingredients submitted: {ingredients}") | |
| return jsonify({'success': True}) | |
| import logging | |
| # Add this to the beginning of your Flask app file | |
| logging.basicConfig(level=logging.DEBUG) # Enable debug-level logging | |
| def get_dish_suggestions(): | |
| selected_ingredients = request.json.get('ingredients', []) | |
| if not selected_ingredients: | |
| return jsonify({'error': 'No ingredients selected'}), 400 | |
| ingredients_text = ', '.join([ingredient['name'] for ingredient in selected_ingredients]) | |
| prompt = f"Suggest a recipe based on the following ingredients: {ingredients_text}" | |
| try: | |
| # Use the correct API structure for openai>=1.0.0 | |
| response = openai.Completion.create( | |
| model="gpt-3.5-turbo", | |
| prompt=prompt, | |
| max_tokens=150, | |
| temperature=0.7 | |
| ) | |
| dish_suggestions = response['choices'][0]['text'].strip() | |
| return jsonify({"suggestions": dish_suggestions}) | |
| except Exception as e: | |
| logging.error(f"Error in /get_dish_suggestions: {e}") | |
| return jsonify({"error": f"Unexpected error: {str(e)}"}), 500 | |
| if __name__ == '__main__': | |
| app.run(debug=True, host='0.0.0.0', port=7860) |