Update src/agent.py

src/agent.py

@@ -99,6 +99,20 @@ class BasicAgent():
         if file_path.endswith(".txt"):
             with open(file_path, "r", encoding="utf-8") as f:
                 return f.read()
+        elif file_path.endswith(".csv"):
+            data = []
+            try:
+                # Use csv.DictReader to read the file into a list of dictionaries
+                with open(file_path, mode='r', newline='', encoding='utf-8') as file:
+                    reader = csv.DictReader(file)
+                    for row in reader:
+                        data.append(row)
+                
+                # Return the structured data as a JSON string
+                return json.dumps(data, indent=2)
+            except Exception as e:
+                print(f"Error reading CSV file: {e}")
+                return ""
         elif file_path.endswith(".docx"):
             doc = docx.Document(file_path)
             return "\n".join([p.text for p in doc.paragraphs])