Update src/text_processor.py

src/text_processor.py

@@ -62,7 +62,15 @@ def generate_logic(llm_output: str) -> str:
     # Extract JSON from the generated text
     start_index = generated_text.find('{')
     end_index = generated_text.rfind('}') + 1
-    return generated_text[start_index:end_index]
+    json_str = generated_text[start_index:end_index]
+    
+    # Log the generated JSON string for debugging
+    print(f"Generated JSON: {json_str}")
+
+    if not json_str.strip():
+        raise ValueError("Generated logic is empty or invalid JSON")
+    
+    return json_str
 
 # Pydantic model for structured output
 class VideoAnalysis(BaseModel):
@@ -73,7 +81,11 @@ class VideoAnalysis(BaseModel):
 
     @classmethod
     def from_llm_output(cls, generated_logic: str) -> 'VideoAnalysis':
-        logic_dict = json.loads(generated_logic)
+        try:
+            logic_dict = json.loads(generated_logic)
+        except json.JSONDecodeError as e:
+            raise ValueError(f"Error decoding JSON: {e}") from e
+        
         return cls(
             screen_interaction_yes=logic_dict.get("Screen.interaction_yes", 0),
             hands_free=logic_dict.get("Hands.free", 0),