Create app.py

app.py

@@ -0,0 +1,79 @@
+from flask import Flask, render_template, request, jsonify
+import yaml
+import json
+try:
+    from src.retrieval import RetrievalSystem
+    from src.generation import ResponseGenerator
+    from src.history import HistoryManager
+except ImportError as e:
+    print(f"Dependency error: {e}. Please ensure all dependencies are installed (pip install -r requirements.txt).")
+    raise
+
+app = Flask(__name__)
+
+# Load configuration
+try:
+    with open("config/config.yaml", "r") as f:
+        config = yaml.safe_load(f)
+except FileNotFoundError:
+    print("Configuration file config.yaml not found in config/ directory.")
+    raise
+
+# Initialize components
+try:
+    retrieval_system = RetrievalSystem(config["data"]["document_path"], config["retrieval"]["model"])
+    response_generator = ResponseGenerator(config["generation"]["model_name"])
+    history_manager = HistoryManager(config["data"]["history_path"])
+except Exception as e:
+    print(f"Initialization error: {e}")
+    raise
+
+# Quick replies for Rupeia categories
+QUICK_REPLIES = [
+    {"title": "Investments", "value": "Tell me about investments"},
+    {"title": "Goals", "value": "How can I set financial goals?"},
+    {"title": "Benefits", "value": "What are the benefits of Rupeia?"},
+    {"title": "Gadgets", "value": "Help me save for a gadget"},
+    {"title": "News & Blogs", "value": "Show me financial news"}
+]
+
+@app.route('/')
+def index():
+    return render_template('index.html')
+
+@app.route('/api/chat', methods=['POST'])
+def chat():
+    try:
+        data = request.get_json()
+        user_input = data.get('message', '').strip()
+        if not user_input:
+            return jsonify({"text": "Please provide a message.", "quickReplies": []}), 400
+
+        # Load history
+        history = history_manager.load_history()
+
+        # Handle welcome message
+        if user_input.lower() == "hello":
+            response = "Hi there! How can I help you today?"
+            history.append(("hello", response))
+            history_manager.save_history(history)
+            return jsonify({"text": response, "quickReplies": QUICK_REPLIES})
+
+        # Retrieve context using RAG
+        context = retrieval_system.get_context(user_input)
+
+        # Generate response
+        response = response_generator.generate_response(user_input, context, history)
+
+        # Update history
+        history.append((user_input, response))
+        history_manager.save_history(history)
+
+        # Return response with quick replies
+        return jsonify({"text": response, "quickReplies": QUICK_REPLIES})
+
+    except Exception as e:
+        return jsonify({"text": f"Sorry, we're experiencing technical difficulties: {str(e)}", "quickReplies": []}), 500
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=8000, debug=True)