Update Lab_report_analysis.py

Lab_report_analysis.py

@@ -1,72 +1,72 @@
-import base64
-import os
-from huggingface_hub import InferenceClient
-from tkinter import Tk, filedialog
-
-def select_image():
-    
-    root = Tk()
-    root.withdraw() 
-    file_path = filedialog.askopenfilename(
-        title="Select a Lab Report Image",
-        filetypes=[("Image Files", "*.jpg;*.jpeg;*.png;*.bmp;*.tiff;*.webp")]
-    )
-    return file_path
-
-
-client = InferenceClient(
-    provider="nebius",
-    api_key=os.getenv("HUGGINGFACE_API_KEY", "your-api-key-here"),
-)
-img = select_image()
-
-with open(img, "rb") as f:
-    image_bytes = f.read()
-    image_b64 = base64.b64encode(image_bytes).decode("utf-8")
-
-prompt = """
-You are a medical analysis assistant.
-
-Analyze the following lab report image and give a structured, professional summary
-following these steps:
-
-1. Extract the results (with normal ranges if available).
-2. Highlight abnormal values clearly.
-3. Explain what the results suggest in simple terms.
-4. Provide an overall summary of health findings.
-5. End with the disclaimer:
-   "This analysis is for educational purposes only and should not replace professional medical advice."
-
-If the image is unreadable, respond: "The image text is unclear."
-"""
-
-completion = client.chat.completions.create(
-    model="google/gemma-3-27b-it",
-    messages=[
-        {
-            "role": "user",
-            "content": [
-                {"type": "text", "text": """
-Analyze this lab report and give me a brief, structured summary.
-
-Format your response as follows:
-
-Summary: (2–3 sentences explaining what the report shows)
-Key Findings: (3–5 bullet points with main abnormal or notable values)
-Interpretation: (1–2 sentences explaining what the findings suggest)
-Note: (One line disclaimer that it’s not medical advice)
-
-Keep it short, clear, and professional — like a medical summary written for quick review.
-"""},
-                {
-                    "type": "image_url",
-                    "image_url": {
-                        "url": f"data:image/jpeg;base64,{image_b64}"
-                    }
-                }
-            ]
-        }
-    ],
-)
-
-print(completion.choices[0].message.content.strip())
+import base64
+import os
+from huggingface_hub import InferenceClient
+from tkinter import Tk, filedialog
+
+def select_image():
+    
+    root = Tk()
+    root.withdraw() 
+    file_path = filedialog.askopenfilename(
+        title="Select a Lab Report Image",
+        filetypes=[("Image Files", "*.jpg;*.jpeg;*.png;*.bmp;*.tiff;*.webp")]
+    )
+    return file_path
+
+
+client = InferenceClient(
+    provider="nebius",
+    api_key=os.getenv("HUGGINGFACE_API_KEY", "your-api-key-here"),
+)
+img = select_image()
+
+with open(img, "rb") as f:
+    image_bytes = f.read()
+    image_b64 = base64.b64encode(image_bytes).decode("utf-8")
+
+prompt = """
+You are a medical analysis assistant.
+
+Analyze the following lab report image and give a structured, professional summary
+following these steps:
+
+1. Extract the results (with normal ranges if available).
+2. Highlight abnormal values clearly.
+3. Explain what the results suggest in simple terms.
+4. Provide an overall summary of health findings.
+5. End with the disclaimer:
+   "This analysis is for educational purposes only and should not replace professional medical advice."
+
+If the image is unreadable, respond: "The image text is unclear."
+"""
+
+completion = client.chat.completions.create(
+    model="google/gemma-2b-it",
+    messages=[
+        {
+            "role": "user",
+            "content": [
+                {"type": "text", "text": """
+Analyze this lab report and give me a brief, structured summary.
+
+Format your response as follows:
+
+Summary: (2–3 sentences explaining what the report shows)
+Key Findings: (3–5 bullet points with main abnormal or notable values)
+Interpretation: (1–2 sentences explaining what the findings suggest)
+Note: (One line disclaimer that it’s not medical advice)
+
+Keep it short, clear, and professional — like a medical summary written for quick review.
+"""},
+                {
+                    "type": "image_url",
+                    "image_url": {
+                        "url": f"data:image/jpeg;base64,{image_b64}"
+                    }
+                }
+            ]
+        }
+    ],
+)
+
+print(completion.choices[0].message.content.strip())