Update src/main.py

src/main.py

@@ -221,20 +221,20 @@ class AICoScientist:
             for j, hypothesis in enumerate(refined_hypotheses):
                 # Get feedback from reflection agent
                 feedback = reflection_agent.review_hypothesis(
-                    hypothesis["hypothesis"],
+                    hypothesis.get("hypothesis", hypothesis.get("statement")),
                     self.research_goal
                 )
                 
                 # Evolve hypothesis based on feedback
                 evolved = evolution_agent.evolve_hypothesis(
-                    hypothesis["hypothesis"],
+                    hypothesis.get("hypothesis", hypothesis.get("statement")),
                     feedback,
                     self.research_goal
                 )
                 
                 # Check proximity to research goal
                 proximity_result = proximity_agent.evaluate_proximity(
-                    evolved["hypothesis"],
+                    evolved.get("hypothesis", evolved.get("statement")),
                     self.research_goal
                 )