Update crew.py

crew.py

@@ -193,14 +193,6 @@ def run_crew(question, file_path):
            Raises:
                RuntimeError: If processing fails"""
         try:
-            #if file_path:
-            #    df = read_file(file_path)
-            #    question = f"Content of file {file_path}:\n{df.to_string()}\nQuestion: {question}\nDO NOT READ the file, but use the content provided above."
-
-            print("###")
-            print(question)
-            print("###")
-            
             client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
 
             response = client.models.generate_content(
@@ -366,9 +358,17 @@ def run_crew(question, file_path):
     # Process
 
     if file_path:
-        question = f"{question} File path: {file_path}."
-        #df = read_file(file_path)
-        #question = f"Content of file {file_path}:\n{df.to_string()}\nQuestion: {question}\nDO NOT READ the file, but use the content provided above."
+        ext = os.path.splitext(file_path)[1].lower()
+
+        if ext in (".csv", ".xls", ".xlsx", ".json", ".jsonl"):
+            df = read_file(file_path)
+            question = f"Content of file {file_path}:\n{df.to_string()}\nQuestion: {question}"
+        else:
+            question = f"{question} File path: {file_path}."
+
+    print("###")
+    print(question)
+    print("###")
     
     initial_answer = crew.kickoff(inputs={"question": question})
     final_answer = get_final_answer(FINAL_ANSWER_MODEL, question, str(initial_answer))