Update app.py

app.py

@@ -4,7 +4,7 @@ from typing import Dict, Any
 from functools import partial
 import warnings
 
-from flask import Flask, request, jsonify
+from flask import Flask, request, jsonify, render_template
 from transformers import pipeline
 
 # Suppress the FutureWarning
@@ -158,6 +158,10 @@ class EliteDeveloperCluster:
 # Flask App for handling agent requests
 app = Flask(__name__)
 
+@app.route('/')
+def index():
+    return render_template('index.html')
+
 @app.route('/agent', methods=['POST'])
 async def agent_request():
     data = request.get_json()
@@ -169,6 +173,16 @@ async def agent_request():
     else:
         return jsonify({'response': 'Invalid input'})
 
+@app.route('/chat', methods=['POST'])
+async def chat():
+    data = request.get_json()
+    query = data.get('query')
+    if query:
+        response = await get_response(query)
+        return jsonify({'response': response})
+    else:
+        return jsonify({'response': 'Invalid input'})
+
 # Chat Interface
 async def get_response(query: str) -> str:
     return await cluster.route_request(query)