Update main.py

main.py

@@ -1,109 +1,31 @@
-import os
-import sys
-import json
-from dotenv import load_dotenv
-import base64
-# from PIL import Image
-from groq import Groq
-# import requests
-
-load_dotenv()
-groq_api_key = os.getenv("GROQ_API_KEY")
-
-# def get_image():
-#     """
-#     Get the image from the specified path.
-#     :return: Image object.
-#     """
-#     image_path = "image/boil_1.jpeg"
-#     if not os.path.exists(image_path):
-#         print(f"Image file {image_path} does not exist.")
-#         sys.exit(1)
-    
-#     image = Image.open(image_path)
-#     return image
-
-# def get_image_path():
-#     """
-#     Get the image path from the specified environment variable.
-#     :return: Image path.
-#     """
-#     image_path = os.getenv("IMAGE_PATH")
-#     if not image_path:
-#         print("IMAGE_PATH environment variable is not set.")
-#         sys.exit(1)
-    
-#     if not os.path.exists(image_path):
-#         print(f"Image file {image_path} does not exist.")
-#         sys.exit(1)
-    
-#     return image_path
-
-
-image_path = "image/boil_1.jpeg"
-if not os.path.exists(image_path):
-    print(f"Image file {image_path} does not exist.")
-    sys.exit(1)
-
-model = "meta-llama/llama-4-scout-17b-16e-instruct"
-query = "What is the diagnosis of this image?"
-# query = "What is the diagnosis of this image? Please provide a detailed explanation of the findings and any relevant differential diagnoses."
-
-
-def encode_image_to_base64(image_path):
-    """
-    Encode an image to base64 format.
-    :param image_path: Path to the image file.
-    :return: Base64 encoded string of the image.
-    """
-    with open(image_path, "rb") as image_file:
-        encoded_string = base64.b64encode(image_file.read()).decode('utf-8')
-    return encoded_string
-
-
-def analyze_image_with_query(query, model, encoded_image):
-    client=Groq()  
-    messages=[
-        {
-            "role": "user",
-            "content": [
-                {
-                    "type": "text", 
-                    "text": query
-                },
-                {
-                    "type": "image_url",
-                    "image_url": {
-                        "url": f"data:image/jpeg;base64,{encoded_image}",
-                    },
-                },
-            ],
-        }]
-    chat_completion=client.chat.completions.create(
-        messages=messages,
-        model=model
-    )
-
-    return chat_completion.choices[0].message.content
-
-# def main():
-#     """
-#     Main function to run the image analysis.
-#     """
-#     # Get the image path from the environment variable
-#     # image_path = get_image_path()
-
-#     # Get the image
-#     # image = get_image()
-
-#     # Encode the image to base64
-#     encoded_image = encode_image_to_base64(image_path)
-
-#     # Analyze the image with the query
-#     response = analyze_image_with_query(query, model, encoded_image)
-
-#     # Print the response
-#     print(response)
-
-# if __name__ == "__main__":
-#     main()
+
+import base64
+from groq import Groq
+
+model = "meta-llama/llama-4-scout-17b-16e-instruct"
+query = "What is the diagnosis of this image?"
+# query = "What is the diagnosis of this image? Please provide a detailed explanation of the findings and any relevant differential diagnoses."
+
+def encode_image_to_base64(image_path):
+    with open(image_path, "rb") as image_file:
+        return base64.b64encode(image_file.read()).decode("utf-8")
+
+def analyze_image_with_query(query, model, encoded_image):
+    client = Groq()
+    messages = [
+        {
+            "role": "user",
+            "content": [
+                {"type": "text", "text": query},
+                {
+                    "type": "image_url",
+                    "image_url": {"url": f"data:image/jpeg;base64,{encoded_image}"}
+                },
+            ],
+        }
+    ]
+    chat_completion = client.chat.completions.create(
+        messages=messages,
+        model=model
+    )
+    return chat_completion.choices[0].message.content