Correct prompt

topic_extraction.py

@@ -267,8 +267,22 @@ def call_gemini_for_table_classification(image_data: bytes, api_key: str) -> str
     except Exception as e:
         logger.warning(f"shrink_image_to_jpeg error: {e}")
 
-    prompt = """You are given an image. Determine if it shows a table that has exactly 2 or 3 columns.
-Return only one label:
+    prompt = """You are given an image. Determine if it shows a table that has exactly 2 or 3 columns. 
+The three-column 'table' image include such key features:
+    - Three columns header columns
+    - Headers like 'Topics', 'Content', 'Guidelines'
+    - Numbered sections (e.g., 8.4, 9.1)
+    - Educational curriculum-style structure
+The two-column 'table' image include such key features:
+    - Two columns header columns
+    - Headers like 'Subject content' and 'Additional information'
+    - Numbered sections (e.g., 2.1, 3.4)
+    - Educational curriculum-style structure
+    - Bullet description in 'Additional information'
+If the image is a relevant table with 2 columns, respond with 'TWO_COLUMN'.
+If the image is a relevant table with 3 columns, respond with 'THREE_COLUMN'.
+If the image does not show a table at all, respond with 'NO_TABLE'.
+Return only one of these exact labels as your entire response:
 TWO_COLUMN
 THREE_COLUMN
 NO_TABLE