Update medicine.py

medicine.py

@@ -42,7 +42,7 @@ def find_relevant_file(topic: str) -> str:
         return None
         
     try:
-        model = genai.GenerativeModel('gemini-2.5-flash-lite')
+        model = genai.GenerativeModel(os.getenv("MODEL","gemini-2.5-flash-lite"))
         prompt = f"""
         From the following list of files, which one is most relevant for: "{topic}"?
         Respond with ONLY the filename, nothing else.
@@ -111,7 +111,7 @@ def handle_query():
                 
                 # Process image with Gemini Vision
                 img = Image.open(file.stream)
-                vision_model = genai.GenerativeModel('gemini-2.5-flash-lite')
+                vision_model = genai.GenerativeModel(os.getenv("MODEL","gemini-2.5-flash-lite"))
                 
                 vision_prompt = [
                     """Identify the medicine from this image. Look for:
@@ -134,7 +134,7 @@ def handle_query():
     # If no medicine identified from image, extract from query text
     if not medicine_topic:
         try:
-            model = genai.GenerativeModel('gemini-2.5-flash-lite')
+            model = genai.GenerativeModel(os.getenv("MODEL","gemini-2.5-flash-lite"))
             extract_prompt = f"""
             From this query: "{user_query}"
             Extract the main medicine or medical topic being asked about.