Update app.py

app.py

@@ -45,6 +45,10 @@ class GroqLLM:
             error_msg = f"Error generating response: {str(e)}"
             print(error_msg)
             return error_msg
+            
+    def generate(self, prompt: Union[str, dict, List[Dict]]) -> str:
+        """Add generate method to make compatible with smolagents CodeAgent"""
+        return self.__call__(prompt)
 
 class DataAnalysisAgent(CodeAgent):
     """Extended CodeAgent with dataset awareness"""
@@ -290,4 +294,4 @@ def main():
         st.error(f"An error occurred: {str(e)}")
 
 if __name__ == "__main__":
-    main()
+    main()