ysharma HF staff commited on
Commit
3bdea3c
·
verified ·
1 Parent(s): ca7ad53

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -4
app.py CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  from crewai import Agent, Task, Crew
2
  import gradio as gr
3
  import asyncio
@@ -100,6 +103,8 @@ class ArticleCrew:
100
 
101
  return [planner_task, writer_task, editor_task]
102
 
 
 
103
  async def process_article(self, topic: str) -> Generator[List[Dict], None, None]:
104
  def add_agent_messages(agent_name: str, tasks: str, emoji: str = "🤖"):
105
  # Add agent header
@@ -148,8 +153,8 @@ class ArticleCrew:
148
 
149
  # Handle the output based on current agent
150
  if self.current_agent == "Editor":
151
- # Not going to show editor's output with metadata
152
- # Instead, I am showing assistant message as the final article
153
  self.message_queue.add_message({
154
  "role": "assistant",
155
  "content": "Final article is ready!",
@@ -242,8 +247,9 @@ class ArticleCrew:
242
  "metadata": {"title": "❌ Error"}
243
  }]
244
 
 
 
245
 
246
- # Gradio code
247
  def create_demo():
248
  article_crew = None
249
 
@@ -263,7 +269,7 @@ def create_demo():
263
  type="messages",
264
  show_label=True,
265
  visible=False,
266
- bubble_full_width=False, # Allow messages to wrap naturally
267
  render_markdown=True # Enable markdown rendering
268
  )
269
 
@@ -317,6 +323,7 @@ def create_demo():
317
 
318
  openai_api_key.submit(show_interface, None, [openai_api_key, chatbot, topic, btn])
319
  btn.click(process_input, [topic, chatbot, openai_api_key], [chatbot])
 
320
 
321
  return demo
322
 
 
1
+ #just formatting things are left now!!
2
+ # i think we are there now
3
+
4
  from crewai import Agent, Task, Crew
5
  import gradio as gr
6
  import asyncio
 
103
 
104
  return [planner_task, writer_task, editor_task]
105
 
106
+ # [Previous code remains the same until the callback functions]
107
+
108
  async def process_article(self, topic: str) -> Generator[List[Dict], None, None]:
109
  def add_agent_messages(agent_name: str, tasks: str, emoji: str = "🤖"):
110
  # Add agent header
 
153
 
154
  # Handle the output based on current agent
155
  if self.current_agent == "Editor":
156
+ # Don't show editor's output with metadata
157
+ # Instead, show completion message and final article
158
  self.message_queue.add_message({
159
  "role": "assistant",
160
  "content": "Final article is ready!",
 
247
  "metadata": {"title": "❌ Error"}
248
  }]
249
 
250
+ # [Rest of the code remains the same]
251
+
252
 
 
253
  def create_demo():
254
  article_crew = None
255
 
 
269
  type="messages",
270
  show_label=True,
271
  visible=False,
272
+ avatar_images=(None, "https://avatars.githubusercontent.com/u/170677839?v=4"),
273
  render_markdown=True # Enable markdown rendering
274
  )
275
 
 
323
 
324
  openai_api_key.submit(show_interface, None, [openai_api_key, chatbot, topic, btn])
325
  btn.click(process_input, [topic, chatbot, openai_api_key], [chatbot])
326
+ topic.submit(process_input, [topic, chatbot, openai_api_key], [chatbot])
327
 
328
  return demo
329