ea4all-gradio-agents-mcp-hackathon-tools-refactoring-vqa

ea4all/ea4all_mcp.py

@@ -115,12 +115,14 @@ async def run_vqa_agentic_system(question: str, diagram: str, request: gr.Reques
 
             with Image.open(diagram) as diagram_:
                 if diagram_.format not in allowed_file_types:
-                    chat_memory.append(ChatMessage(role="assistant", content="Invalid file type. Allowed file types are JPEG and PNG."))
-                else:
-                    #'vqa_image = e4u.get_raw_image(diagram) #MOVED into Graph
-                    vqa_image = diagram
-                    response = await diagram_graph.ainvoke({"question":msg, "image": vqa_image}, config)
-                    chat_memory.append(ChatMessage(role="assistant", content=response['messages'][-1].content))
+                    #chat_memory.append(ChatMessage(role="assistant", content="Invalid file type. Allowed file types are JPEG and PNG."))
+                    print(f"---DIAGRAM: {diagram.format} is not a valid file type. Allowed file types are JPEG and PNG.---")
+                #else:
+                #'vqa_image = e4u.get_raw_image(diagram) #MOVED into Graph
+
+                vqa_image = diagram
+                response = await diagram_graph.ainvoke({"question":msg, "image": vqa_image}, config)
+                chat_memory.append(ChatMessage(role="assistant", content=response['messages'][-1].content))
                 
                 yield chat_memory