Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -143,10 +143,10 @@ class AgentLoader:
|
|
| 143 |
return DocsReader
|
| 144 |
|
| 145 |
async def writerAgent():
|
| 146 |
-
writer=Agent(
|
| 147 |
role='Content Writer',
|
| 148 |
goal='To produce higly accurate and easy to understand information',
|
| 149 |
-
backstory="""You are
|
| 150 |
You should provide indetail results on the summary data.""",
|
| 151 |
verbose=True,
|
| 152 |
llm=llm
|
|
@@ -156,16 +156,18 @@ class AgentLoader:
|
|
| 156 |
#<------------------------------Agents END------------------------->
|
| 157 |
|
| 158 |
#<-------------------------------Tasks---------------------------->
|
| 159 |
-
def getTasks(query, agent, exp):
|
| 160 |
-
task_read=Task(
|
| 161 |
description=f'{query}',
|
| 162 |
agent=agent,
|
| 163 |
expected_output=f'A detailed information on {query}'
|
| 164 |
)
|
| 165 |
|
| 166 |
-
|
|
|
|
|
|
|
| 167 |
description=f'{query}',
|
| 168 |
-
agent=
|
| 169 |
expected_output=exp
|
| 170 |
)
|
| 171 |
|
|
@@ -174,29 +176,32 @@ def getTasks(query, agent, exp):
|
|
| 174 |
# Gradio interface function
|
| 175 |
def process_file(file, query, expected_output):
|
| 176 |
path = file.name
|
| 177 |
-
|
| 178 |
-
|
| 179 |
-
|
| 180 |
-
|
| 181 |
-
|
| 182 |
-
|
| 183 |
-
|
| 184 |
-
|
| 185 |
-
|
| 186 |
-
|
| 187 |
-
|
| 188 |
-
|
| 189 |
-
|
| 190 |
-
|
| 191 |
-
|
|
|
|
| 192 |
mycrew = Crew(
|
| 193 |
-
agents=[agent, AgentLoader.writerAgent()],
|
| 194 |
-
tasks=
|
| 195 |
verbose=True
|
| 196 |
)
|
| 197 |
results = mycrew.kickoff()
|
| 198 |
-
|
| 199 |
-
|
|
|
|
|
|
|
| 200 |
|
| 201 |
# Create the Gradio interface
|
| 202 |
interface = gr.Interface(
|
|
@@ -213,4 +218,3 @@ interface = gr.Interface(
|
|
| 213 |
|
| 214 |
# Launch the Gradio interface
|
| 215 |
interface.launch()
|
| 216 |
-
|
|
|
|
| 143 |
return DocsReader
|
| 144 |
|
| 145 |
async def writerAgent():
|
| 146 |
+
writer = Agent(
|
| 147 |
role='Content Writer',
|
| 148 |
goal='To produce higly accurate and easy to understand information',
|
| 149 |
+
backstory="""You are a content specialist and are responsible to generate reliable and easy to understand content or information based on the summary of data.
|
| 150 |
You should provide indetail results on the summary data.""",
|
| 151 |
verbose=True,
|
| 152 |
llm=llm
|
|
|
|
| 156 |
#<------------------------------Agents END------------------------->
|
| 157 |
|
| 158 |
#<-------------------------------Tasks---------------------------->
|
| 159 |
+
async def getTasks(query, agent, exp):
|
| 160 |
+
task_read = Task(
|
| 161 |
description=f'{query}',
|
| 162 |
agent=agent,
|
| 163 |
expected_output=f'A detailed information on {query}'
|
| 164 |
)
|
| 165 |
|
| 166 |
+
writer_agent = await AgentLoader.writerAgent()
|
| 167 |
+
|
| 168 |
+
task_write = Task(
|
| 169 |
description=f'{query}',
|
| 170 |
+
agent=writer_agent,
|
| 171 |
expected_output=exp
|
| 172 |
)
|
| 173 |
|
|
|
|
| 176 |
# Gradio interface function
|
| 177 |
def process_file(file, query, expected_output):
|
| 178 |
path = file.name
|
| 179 |
+
|
| 180 |
+
async def process_async():
|
| 181 |
+
if path.endswith(".pdf"):
|
| 182 |
+
agent = await AgentLoader.PdfReaderAgent(path)
|
| 183 |
+
elif path.endswith(".docx"):
|
| 184 |
+
agent = await AgentLoader.DocsReaderAgent(path)
|
| 185 |
+
elif path.endswith(".json") or path.endswith(".txt"):
|
| 186 |
+
agent = await AgentLoader.fileReaderAgent(path)
|
| 187 |
+
elif path.endswith(".csv"):
|
| 188 |
+
agent = await AgentLoader.csvReaderAgent(path)
|
| 189 |
+
results = agent.run(query)
|
| 190 |
+
return results
|
| 191 |
+
else:
|
| 192 |
+
return 'File NOT supported'
|
| 193 |
+
|
| 194 |
+
tasks = await getTasks(query, agent, expected_output)
|
| 195 |
mycrew = Crew(
|
| 196 |
+
agents=[agent, await AgentLoader.writerAgent()],
|
| 197 |
+
tasks=tasks,
|
| 198 |
verbose=True
|
| 199 |
)
|
| 200 |
results = mycrew.kickoff()
|
| 201 |
+
return results
|
| 202 |
+
|
| 203 |
+
loop = asyncio.get_event_loop()
|
| 204 |
+
return loop.run_until_complete(process_async())
|
| 205 |
|
| 206 |
# Create the Gradio interface
|
| 207 |
interface = gr.Interface(
|
|
|
|
| 218 |
|
| 219 |
# Launch the Gradio interface
|
| 220 |
interface.launch()
|
|
|