Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ import google.generativeai as genai
|
|
5 |
|
6 |
# Tool import
|
7 |
from crewai.tools.gemini_tools import GeminiSearchTools
|
8 |
-
|
9 |
|
10 |
# Google Langchain
|
11 |
from langchain_google_genai import GoogleGenerativeAI
|
@@ -58,7 +58,7 @@ def crewai_process(research_topic):
|
|
58 |
allow_delegation=False,
|
59 |
llm = gemini_llm,
|
60 |
tools=[
|
61 |
-
|
62 |
]
|
63 |
|
64 |
# Add tools and other optional parameters as needed
|
@@ -143,14 +143,20 @@ def crewai_process(research_topic):
|
|
143 |
agent=finalizer
|
144 |
)
|
145 |
|
|
|
|
|
|
|
|
|
|
|
|
|
146 |
# Instantiate your crew with a sequential process
|
147 |
crew = Crew(
|
148 |
# agents=[author, artist, poet, reader, finalizer ],
|
149 |
# tasks=[task1, task2, task3, task4, task5]
|
150 |
# agents=[author, artist, poet, finalizer ],
|
151 |
# tasks=[task1, task2, task3, task5],
|
152 |
-
agents=[author,
|
153 |
-
tasks=[task1,
|
154 |
verbose=2,
|
155 |
process=Process.sequential
|
156 |
)
|
@@ -173,6 +179,5 @@ if st.button("Run"):
|
|
173 |
# Run the crewai process
|
174 |
result = crewai_process(input_topic)
|
175 |
# Display the result
|
176 |
-
|
177 |
|
178 |
-
st.image(result, caption='Generated Image', use_column_width=True)
|
|
|
5 |
|
6 |
# Tool import
|
7 |
from crewai.tools.gemini_tools import GeminiSearchTools
|
8 |
+
|
9 |
|
10 |
# Google Langchain
|
11 |
from langchain_google_genai import GoogleGenerativeAI
|
|
|
58 |
allow_delegation=False,
|
59 |
llm = gemini_llm,
|
60 |
tools=[
|
61 |
+
GeminiSearchTools.gemini_search
|
62 |
]
|
63 |
|
64 |
# Add tools and other optional parameters as needed
|
|
|
143 |
agent=finalizer
|
144 |
)
|
145 |
|
146 |
+
task6 = Task(
|
147 |
+
description="""output both the story created by the author""",
|
148 |
+
agent=finalizer
|
149 |
+
|
150 |
+
)
|
151 |
+
|
152 |
# Instantiate your crew with a sequential process
|
153 |
crew = Crew(
|
154 |
# agents=[author, artist, poet, reader, finalizer ],
|
155 |
# tasks=[task1, task2, task3, task4, task5]
|
156 |
# agents=[author, artist, poet, finalizer ],
|
157 |
# tasks=[task1, task2, task3, task5],
|
158 |
+
agents=[author, finalizer],
|
159 |
+
tasks=[task1, task6],
|
160 |
verbose=2,
|
161 |
process=Process.sequential
|
162 |
)
|
|
|
179 |
# Run the crewai process
|
180 |
result = crewai_process(input_topic)
|
181 |
# Display the result
|
182 |
+
st.text_area("Output", value=result, height=300)
|
183 |
|
|