pre-commit checking (#18)

* support pre-commit

* support running

* pre-commit running

* pre-commit running

.gitignore

@@ -157,4 +157,4 @@ cython_debug/
 #  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
-#.idea/
+#.idea/

.pre-commit-config.yaml

@@ -24,4 +24,4 @@ repos:
 -   repo: https://github.com/kynan/nbstripout
     rev: 0.6.0
     hooks:
-      - id: nbstripout
+      - id: nbstripout

app.py

@@ -2,8 +2,8 @@ import os
 import sys
 
 import gradio as gr
-import sys
-sys.path.append('./ctm')
+
+sys.path.append("./ctm")
 from ctm.ctms.ctm_base import BaseConsciousnessTuringMachine
 
 ctm = BaseConsciousnessTuringMachine()
@@ -17,27 +17,27 @@ def convert_base64(image_array):
     buffer = io.BytesIO()
     image.save(buffer, format="PNG")
     byte_data = buffer.getvalue()
-    base64_string = base64.b64encode(byte_data).decode('utf-8')
+    base64_string = base64.b64encode(byte_data).decode("utf-8")
     return base64_string
 
 
 def introduction():
     with gr.Column(scale=2):
-        gr.Image(
-            "images/CTM-AI.png", elem_id="banner-image", show_label=False
-        )
+        gr.Image("images/CTM-AI.png", elem_id="banner-image", show_label=False)
     with gr.Column(scale=5):
         gr.Markdown(
             """Consciousness Turing Machine Demo
             """
         )
 
+
 def add_processor(processor_name, display_name, state):
-    print('add processor ', processor_name)
+    print("add processor ", processor_name)
     ctm.add_processor(processor_name)
     print(ctm.processor_group_map)
     print(len(ctm.processor_list))
-    return display_name + ' (added)'
+    return display_name + " (added)"
+
 
 def processor_tab():
     # Categorized model names
@@ -45,14 +45,14 @@ def processor_tab():
         "gpt4_text_emotion_processor",
         "gpt4_text_summary_processor",
         "gpt4_speaker_intent_processor",
-        "roberta_text_sentiment_processor"
+        "roberta_text_sentiment_processor",
     ]
     vision_processors = [
         "gpt4v_cloth_fashion_processor",
         "gpt4v_face_emotion_processor",
         "gpt4v_ocr_processor",
         "gpt4v_posture",
-        "gpt4v_scene_location_processor"
+        "gpt4v_scene_location_processor",
     ]
 
     with gr.Blocks():
@@ -60,37 +60,53 @@ def processor_tab():
             with gr.Column(scale=1):
                 gr.Markdown("### Text Processors")
                 for model_name in text_processors:
-                    display_name = model_name.replace("processor", "").replace("_", " ").title()
+                    display_name = (
+                        model_name.replace("processor", "")
+                        .replace("_", " ")
+                        .title()
+                    )
 
                     button = gr.Button(display_name)
-                    processor_name = gr.Textbox(value=model_name, visible=False)
-                    display_name = gr.Textbox(value=display_name, visible=False)
+                    processor_name = gr.Textbox(
+                        value=model_name, visible=False
+                    )
+                    display_name = gr.Textbox(
+                        value=display_name, visible=False
+                    )
                     button.click(
                         fn=add_processor,
                         inputs=[processor_name, display_name, gr.State()],
-                        outputs=[button]
+                        outputs=[button],
                     )
 
             with gr.Column(scale=1):
                 gr.Markdown("### Vision Processors")
                 for model_name in vision_processors:
-                    display_name = model_name.replace("processor", "").replace("_", " ").title()
+                    display_name = (
+                        model_name.replace("processor", "")
+                        .replace("_", " ")
+                        .title()
+                    )
 
                     button = gr.Button(display_name)
-                    processor_name = gr.Textbox(value=model_name, visible=False)
-                    display_name = gr.Textbox(value=display_name, visible=False)
+                    processor_name = gr.Textbox(
+                        value=model_name, visible=False
+                    )
+                    display_name = gr.Textbox(
+                        value=display_name, visible=False
+                    )
                     button.click(
                         fn=add_processor,
                         inputs=[processor_name, display_name, gr.State()],
-                        outputs=[button]
+                        outputs=[button],
                     )
 
 
-
-
 def forward(query, content, image, state):
-    state['question'] = query
-    ask_processors_output_info, state = ask_processors(query, content, image, state)
+    state["question"] = query
+    ask_processors_output_info, state = ask_processors(
+        query, content, image, state
+    )
     uptree_competition_output_info, state = uptree_competition(state)
     ask_supervisor_output_info, state = ask_supervisor(state)
 
@@ -148,8 +164,8 @@ def interface_tab():
             text = gr.Textbox(label="Enter your text here")
             query = gr.Textbox(label="Enter your query here")
             image = gr.Image(label="Upload your image")
-            #audio = gr.Audio(label="Upload or Record Audio")
-            #video = gr.Video(label="Upload or Record Video")
+            # audio = gr.Audio(label="Upload or Record Audio")
+            # video = gr.Video(label="Upload or Record Video")
 
             # Processing buttons
             forward_button = gr.Button("Start CTM forward process")

ctm/ctms/ctm_base.py

@@ -207,7 +207,9 @@ class BaseConsciousnessTuringMachine(object):
                 audio_path=audio_path,
                 video_path=video_path,
             )
-            import pdb; pdb.set_trace()
+            import pdb
+
+            pdb.set_trace()
             winning_output = self.uptree_competition(processor_output)
             answer, score = self.ask_supervisor(question, winning_output)
             if score > answer_threshold:

requirements.txt

@@ -2,4 +2,5 @@ scikit-learn==1.3.0
 huggingface-hub==0.21.4
 gradio==4.27.0
 gradio_client==0.15.1
-openai==1.23.3
+openai==1.23.3
+pre-commit==3.7.0