Update main.py

main.py

@@ -1,7 +1,7 @@
-from flask import Flask, jsonify
+from flask import Flask, jsonify, request
 import asyncio
 import os
-from hiou import hugging, Ahugging, browser
+from hiou import Ahugging, browser
 import json
 
 async def get_answer(image: str, question: str):
@@ -35,11 +35,20 @@ async def get_answer(image: str, question: str):
 
 app = Flask(__name__)
 
-# Example async route using asyncio.sleep
-@app.route('/async-task')
+@app.route('/getans', methods=['POST'])
 async def async_task():
-    await asyncio.sleep(2)  # Simulate async task
-    return jsonify({"message": "Async task completed!"})
+    if 'file' not in request.files:
+        jsonify({"status" : False, "msg" : "No file found"}), 400
+    file = request.files['file']
+    if file.filename == '':
+        return jsonify({"error": "No selected file"}), 400
+    file_content = await file.read()
+    headers = dict(request.headers)
+    if not headers.get("KEY") == os.getenv("KEY"):
+        return jsonify({"status" : False, "msg" : "Invalid API Key"}), 404
+    print(f"QUESTION ASKED : {headers.get('QU', '')}")
+    answer = await get_answer(file_content, headers.get('QU', ''))
+    return jsonify({"status" : True, "ANS" : answer})
 
 @app.route('/')
 def home():