Update app.py

app.py

@@ -1290,9 +1290,11 @@ If you find any "Code-Blocks" then,
     ]
 
     try:
+        return 
         # Invoke the main agent for logic refinement and relationship identification
-        response = agent.invoke({"messages": [{"role": "user", "content": content}]})
-        llm_output_raw = response["messages"][-1].content.strip()
+        # response = agent.invoke({"messages": [{"role": "user", "content": content}]})
+        # llm_output_raw = response["messages"][-1].content.strip()
+        llm_output_raw = ""
         print(f"llm_output_raw: {response}")
         parsed_llm_output = extract_json_from_llm_response(llm_output_raw)        
         result = parsed_llm_output           
@@ -1313,8 +1315,9 @@ If you find any "Code-Blocks" then,
             "Corrected JSON:\n"
         )
         try:
-            correction_response = agent_json_resolver.invoke({"messages": [{"role": "user", "content": correction_prompt}]})
-            corrected_output = extract_json_from_llm_response(correction_response['messages'][-1].content)
+            # correction_response = agent_json_resolver.invoke({"messages": [{"role": "user", "content": correction_prompt}]})
+            # corrected_output = extract_json_from_llm_response(correction_response['messages'][-1].content)
+            corrected_output = extract_json_from_llm_response(llm_output_raw)
             #block_relationships = corrected_output.get("block_relationships", [])
             result = corrected_output 
             print(f"result:\n\n {result}")