cat-llm 0.0.25__py3-none-any.whl → 0.0.26__py3-none-any.whl

{cat_llm-0.0.25.dist-info → cat_llm-0.0.26.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cat-llm
-Version: 0.0.25
+Version: 0.0.26
 Summary: A tool for categorizing text data and images using LLMs and vision models
 Project-URL: Documentation, https://github.com/chrissoria/cat-llm#readme
 Project-URL: Issues, https://github.com/chrissoria/cat-llm/issues

cat_llm-0.0.26.dist-info/RECORD

@@ -0,0 +1,9 @@
+catllm/CERAD_functions.py,sha256=mtHxshRWmWXMH9kkkCfbMHXgDe00EVabjFiN8s73LPI,16935
+catllm/__about__.py,sha256=4jqHFlCUy8Xa_lkK5V6tmc1z_Goja59oOIaD7pFXnRs,404
+catllm/__init__.py,sha256=kLk180aJna1s-wU6CLr4_hKkbjoeET-11jGmC1pdhQw,330
+catllm/cat_llm.py,sha256=Rwyz93caNf0h9tfurObY6qDjtG6EKaYXR0GrVW7h2kU,16920
+catllm/image_functions.py,sha256=nTppbeeuYrlDmZtUkEwNwRJQ6YbBxBCYoPRU1X4GZOY,27100
+cat_llm-0.0.26.dist-info/METADATA,sha256=uXjULRmbv-RQu9Q1aU3rXuYo1d4qK-163M2i-j5Sb1o,1679
+cat_llm-0.0.26.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+cat_llm-0.0.26.dist-info/licenses/LICENSE,sha256=wJLsvOr6lrFUDcoPXExa01HOKFWrS3JC9f0RudRw8uw,1075
+cat_llm-0.0.26.dist-info/RECORD,,

catllm/CERAD_functions.py

@@ -2,6 +2,12 @@
 # specifically for pictures of drawings of shapes like circles, diamonds, rectangles, and cubes
 
 """
+Ket features:
+1. Shape-specific scoring: The function can handle different shapes (circle, diamond, rectangles, cube) and provides tailored categories for each shape.
+2. Image input handling: It accepts image inputs either as file paths or a list of images.
+3. Model flexibility: The function allows users to specify different models (OpenAI, Anthropic, Perplexity, Mistral) for image analysis.
+4. Safety and progress saving: It can save progress to a CSV file, which is useful for long-running tasks or when processing many images.
+
 Areas for improvement:
 1. Prompt refinement: adjusting the prompt so that it produces a more accurate score.
 2. Image preprocessing: adjusting the images so that they are easier to be analyzed by the models.
@@ -19,7 +25,7 @@ def cerad_drawn_score(
     shape, 
     image_input,
     api_key,
-    user_model="gpt-4o-2024-11-20",
+    user_model="gpt-4o",
     creativity=0,
     reference_in_image=False,
     provide_reference=False,
@@ -40,8 +46,8 @@ def cerad_drawn_score(
 
     if shape == "circle":
         categories = ["The image contains a drawing that clearly represents a circle",
-                    "The drawing does not resemble a circle",
-                    "The drawing resembles a circle",
+                    "The image does NOT contain any drawing that resembles a circle",
+                    "The image contains a drawing that resembles a circle",
                     "The circle is closed",
                     "The circle is almost closed",
                     "The circle is circular",
@@ -58,12 +64,12 @@ def cerad_drawn_score(
                     "None of the above descriptions apply"]
     elif shape == "rectangles" or shape == "overlapping rectangles":
         categories = ["The image contains a drawing that clearly represents overlapping rectangles",
-                    "A drawn shape DOES NOT resemble a overlapping rectangles",
-                    "A drawn shape resembles a overlapping rectangles",
-                    "Rectangle 1 has 4 sides",
-                    "Rectangle 2 has 4 sides",
-                    "The rectangles are overlapping",
-                    "The rectangles overlap contains a longer vertical rectangle with top and bottom portruding",
+                    "The image does NOT contain any drawing that resembles overlapping rectangles",
+                    "The image contains a drawing that resembles overlapping rectangles",
+                    "If rectangle 1 is present it has 4 sides",
+                    "If rectablge 2 is present it has 4 sides",
+                    "The drawn rectangles are overlapping",
+                    "The drawn rectangles overlap to form a longer vertical rectangle with top and bottom sticking out",
                     "None of the above descriptions apply"]
     elif shape == "cube":
         categories = ["The image contains a drawing that clearly represents a cube (3D box shape)",

catllm/__about__.py

@@ -1,7 +1,7 @@
 # SPDX-FileCopyrightText: 2025-present Christopher Soria <chrissoria@berkeley.edu>
 #
 # SPDX-License-Identifier: MIT
-__version__ = "0.0.25"
+__version__ = "0.0.26"
 __author__ = "Chris Soria"
 __email__ = "chrissoria@berkeley.edu"
 __title__ = "cat-llm"

catllm/__init__.py

@@ -12,4 +12,5 @@ from .__about__ import (
 )
 
 from .cat_llm import *
-from .CERAD_functions import * 
+from .CERAD_functions import * 
+from .image_functions import *