Update app.py

app.py

@@ -7,6 +7,7 @@ from crewai_tools import PDFSearchTool, FileReadTool, DOCXSearchTool, CSVSearchT
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain.agents.agent_types import AgentType
 from langchain_experimental.agents.agent_toolkits import create_csv_agent
+import asyncio
 
 # API keys-----------------move them to ENV 
 os.environ["OPENAI_API_KEY"] = "NA"
@@ -90,8 +91,9 @@ class tools:
 
 class AgentLoader:
 
-    def csvReaderAgent(path):
-        agent = create_csv_agent(
+    @staticmethod
+    async def csvReaderAgent(path):
+        agent = await create_csv_agent(
             ChatGoogleGenerativeAI(temperature=0.6, model="gemini-1.5-flash-latest"),
             path,
             verbose=True,
@@ -171,7 +173,7 @@ def getTasks(query, agent, exp):
     return [task_read, task_write]
 
 # Gradio interface function
-def process_file(file, query, expected_output):
+async def process_file(file, query, expected_output):
     path = file.name
     
     if path.endswith(".pdf"):
@@ -181,8 +183,8 @@ def process_file(file, query, expected_output):
     elif path.endswith(".json") or path.endswith(".txt"):
         agent = AgentLoader.fileReaderAgent(path)
     elif path.endswith(".csv"):
-        agent = AgentLoader.csvReaderAgent(path)
-        return agent.run(query)
+        agent = await AgentLoader.csvReaderAgent(path)
+        results = agent.run(query)
     else:
         return 'File NOT supported'
     
@@ -199,7 +201,7 @@ def process_file(file, query, expected_output):
 
 # Create the Gradio interface
 interface = gr.Interface(
-    fn=process_file,
+    fn=lambda file, query, expected_output: asyncio.run(process_file(file, query, expected_output)),
     inputs=[
         gr.File(label="Upload File"),
         gr.Textbox(label="Query"),