Update app.py

app.py

@@ -0,0 +1,67 @@
+from flask import Flask, render_template, request, jsonify
+from database import create_database, get_all_users, get_vacation_location
+from langchain_core.prompts import ChatPromptTemplate
+from langchain_ollama.llms import OllamaLLM
+
+# Initialize Flask app
+app = Flask(__name__)
+
+# Initialize the database
+create_database()
+
+# Define the template for generating a vacation-related question
+template = """You are planning a vacation. Based on the location '{location}', what would be a good question to ask the vacationer to learn more about their vacation plans?"""
+
+# Initialize the prompt and model using LangChain
+prompt = ChatPromptTemplate.from_template(template)
+#model = OllamaLLM(model="llama3.1")
+model = OllamaLLM(model="llama3.2:1b")
+# Define the chain using the prompt and model
+chain = prompt | model
+
+@app.route('/')
+def index():
+    """Render the homepage with a dropdown of usernames."""
+    users = get_all_users()
+    return render_template('index.html', users=users)
+
+@app.route('/vacation', methods=['POST'])
+def vacation():
+    """Display the vacation location without AI interaction."""
+    user_id = request.form.get('user_id')
+
+    if not user_id:
+        return "User ID is missing", 400
+
+    location = get_vacation_location(user_id)
+    username = next((user[1] for user in get_all_users() if user[0] == int(user_id)), None)
+
+    if not username:
+        return "User not found", 404
+
+    return render_template('vacation.html', username=username, location=location)
+
+@app.route('/generate_question', methods=['POST'])
+def generate_question():
+    """Generate a vacation-related question using LangChain."""
+    data = request.json  # Read the JSON payload from the POST request
+    print("Received data:", data)  # Debugging
+
+    location = data.get('location')
+    print("Extracted location:", location)  # Debugging
+
+    if not location:
+        return jsonify({"error": "Location is missing"}), 400
+
+    try:
+        # Generate the question using LangChain
+        question = chain.invoke({"location": location})
+        print("Generated question:", question)  # Debugging
+    except Exception as e:
+        print("Error during question generation:", e)  # Debugging
+        return jsonify({"error": f"Failed to generate question: {str(e)}"}), 500
+
+    return jsonify({"question": question}), 200
+
+if __name__ == '__main__':
+    app.run(debug=True)