Update main.py

main.py

@@ -5,8 +5,12 @@ import requests
 from flask import Flask, request, jsonify
 from werkzeug.utils import secure_filename
 import google.generativeai as genai
+from dotenv import load_dotenv
 from flask_cors import CORS
 
+
+# Step 1: API Key aur Environment Setup
+load_dotenv()
 try:
     genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 except TypeError:
@@ -117,7 +121,8 @@ def process_query():
         classification = SESSIONS[session_id].get('classification')
         category = classification['category']
         endpoint_url = API_ENDPOINTS.get(category)
-        response = requests.post(endpoint_url, json={"query": query}) or requests.post(endpoint_url, data={"query": query}) or requests.post(endpoint_url, files={"query": query})
+        payload={"query": query}
+        response = requests.post(endpoint_url,data=payload, json={"main_query": query}) 
         del SESSIONS[session_id]
         print(f"Session {session_id} closed.")
         return jsonify({
@@ -184,3 +189,5 @@ def process_with_image():
         print(f"Session {session_id} closed after failed API call.")
         return jsonify({"status": "error", "message": f"Backend service call failed: {e}"}), 503
 
+if __name__ == '__main__':
+    app.run(debug=True, port=5000)