Update app.py

app.py

@@ -45,7 +45,7 @@ class ModelTrainingInput(ResearchInput):
 class DataAnalyzer(ABC):
     """Abstract base class for data analysis modules"""
     @abstractmethod
-    def invoke(self, **kwargs) -> Dict[str, Any]:
+    def invoke(self, data:pd.DataFrame, **kwargs) -> Dict[str, Any]:
         pass
 
 # ---------------------- Concrete Analyzer Implementations ---------------------------
@@ -323,7 +323,10 @@ class AutomatedInsights():
        for name in analysis_names:
            if name in self.analyses:
              analyzer = self.analyses[name]
-             results[name] = analyzer.invoke(data=data, **kwargs)
+             try:
+                results[name] = analyzer.invoke(data=data, **kwargs)
+             except Exception as e:
+                results[name] = {"error": str(e)}
            else:
                results[name] = {"error": "Analysis not found"}
        return results
@@ -661,7 +664,7 @@ def main():
                           result = st.session_state.kpi_monitoring.calculate_kpis(data)
                           st.json(result)
             with insights_tab:
-              if selected_data_key:
+                if selected_data_key:
                   data = st.session_state.data[selected_data_key]
                   available_analysis = ["EDA", "temporal", "distribution", "hypothesis", "model"]
                   selected_analysis = st.multiselect("Select Analysis", available_analysis)
@@ -685,28 +688,4 @@ def main():
                   if st.button("Generate Treatment Recommendation"):
                       if condition_col and treatment_col:
                           with st.spinner("Generating Treatment Recommendation"):
-                              result = st.session_state.treatment_recommendation.recommend(data, condition_col = condition_col, treatment_col = treatment_col)
-                              st.json(result)
-
-            with reports_tab:
-                 st.header("Reports")
-                 report_name = st.text_input("Report Name")
-                 report_def = st.text_area("Report definition")
-                 if st.button("Create Report Definition"):
-                   st.session_state.automated_reports.create_report_definition(report_name, report_def)
-                   st.success("Report definition created")
-                 if selected_data_key:
-                   data = st.session_state.data
-                   if st.button("Generate Report"):
-                     with st.spinner("Generating Report..."):
-                       report = st.session_state.automated_reports.generate_report(report_name, data)
-            with knowledge_tab:
-              st.header("Medical Knowledge")
-              query = st.text_input("Enter your medical question here:")
-              if st.button("Search"):
-                  with st.spinner("Searching..."):
-                      result = st.session_state.knowledge_base.search_medical_info(query)
-                      st.write(result)
-
-if __name__ == "__main__":
-    main()
+                              result = st.session_state.treatment_recommendation.recommend(data, condition_col = condition_col