Update medicine.py

medicine.py

@@ -2,11 +2,15 @@ import os
 import io
 from flask import Flask, request, jsonify
 from PIL import Image
+from dotenv import load_dotenv
 import google.generativeai as genai
 import json
 
 # --- INITIAL SETUP ---
 
+# Load environment variables from the .env file
+load_dotenv()
+
 # Configure the Gemini API with your key
 api_key = os.getenv("GOOGLE_API_KEY")
 if not api_key:
@@ -17,7 +21,7 @@ genai.configure(api_key=api_key)
 app = Flask(__name__)
 
 # --- CONFIGURATION ---
-TEXT_FILES_DIR = "Text_Files"
+TEXT_FILES_DIR = "MEDICINE_TXT"
 # Allowed file extensions for image uploads
 ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif'}
 
@@ -86,10 +90,11 @@ def handle_query():
     """
     # 1. Get and validate the request data
     form_data = request.form
-    if not form_data or 'query' not in form_data:
+    query = request.json.get('main_query')
+    if not query:
         return jsonify({"error": "Missing 'query' in request"}), 400
 
-    user_query = form_data.get('query')
+    user_query = query
     medicine_topic = None
     
     # 2. Handle File Upload (if provided)