Update disease.py

disease.py

@@ -2,6 +2,9 @@ import os
 import json
 from flask import Flask, request, jsonify
 import google.generativeai as genai
+import dotenv
+
+dotenv.load_dotenv()
 
 # --- Configuration ---
 FACT_SHEET_DIR = "Text_Files"
@@ -86,10 +89,11 @@ def ask_question():
         return jsonify({"error": "Gemini client is not configured. Check your API key."}), 500
 
     data = request.get_json()
-    if not data or 'query' not in data:
+    form_data = request.form
+    if not form_data and 'query' not in data:
         return jsonify({"error": "Request must be JSON and contain a 'query' field."}), 400
 
-    user_query = data['query']
+    user_query = data['query'] or form_data['query']
     print(f"\n=================================================")
     print(f"Received new query: '{user_query}'")
     print(f"=================================================")