Update app.py

app.py

@@ -1,4 +1,3 @@
-# This works even better 
 from crewai import Agent, Task, Crew
 import gradio as gr
 from gradio import ChatMessage
@@ -9,7 +8,7 @@ from typing import List, Generator
 import os
 from dotenv import load_dotenv
 import threading
-
+from langchain_openai import ChatOpenAI
 
 class OutputParser:
     def __init__(self):
@@ -119,8 +118,19 @@ class StreamingCapture:
         pass
 
 class ArticleCrew:
-    def __init__(self):
-        # Initialize agents
+    def __init__(self, api_key: str = None):
+        self.api_key = api_key
+        self.initialize_agents()
+    
+    def initialize_agents(self):
+        # Create a ChatOpenAI instance with the API key
+        llm = ChatOpenAI(
+            openai_api_key=self.api_key,
+            temperature=0.7,
+            model="gpt-4"
+        )
+        
+        # Initialize agents with the LLM
         self.planner = Agent(
             role="Content Planner",
             goal="Plan engaging and factually accurate content on {topic}",
@@ -128,7 +138,8 @@ class ArticleCrew:
                      "You collect information that helps the audience learn something "
                      "and make informed decisions.",
             allow_delegation=False,
-            verbose=True
+            verbose=True,
+            llm=llm
         )
         
         self.writer = Agent(
@@ -137,7 +148,8 @@ class ArticleCrew:
             backstory="You're working on writing a new opinion piece about the topic: {topic}. "
                      "You base your writing on the work of the Content Planner.",
             allow_delegation=False,
-            verbose=True
+            verbose=True,
+            llm=llm
         )
         
         self.editor = Agent(
@@ -145,7 +157,8 @@ class ArticleCrew:
             goal="Edit a given blog post to align with the writing style",
             backstory="You are an editor who receives a blog post from the Content Writer.",
             allow_delegation=False,
-            verbose=True
+            verbose=True,
+            llm=llm
         )
         
         self.output_parser = OutputParser()
@@ -234,30 +247,40 @@ class ArticleCrew:
             sys.stdout = original_stdout
 
 def create_demo():
-    article_crew = ArticleCrew()
+    article_crew = None  # Initialize as None
     
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
         gr.Markdown("# 📝 AI Article Writing Crew")
-        gr.Markdown("Watch as our AI crew collaborates to create your article!")
-        
+        gr.Markdown("Watch as our AI crew collaborates to create your article! This app uses CrewAI agents: Content Planner, Content Writer, and Content Editor to write Aticle on any topic of your choice. To start, enter your OpenAI API Key below and pres Enter!")
+        openai_api_key = gr.Textbox(
+            label='OpenAI API Key', 
+            type='password', 
+            placeholder='Type your OpenAI API key and press Enter!', 
+            interactive=True)
+
         chatbot = gr.Chatbot(
             label="Writing Process",
             avatar_images=(None, "https://avatars.githubusercontent.com/u/170677839?v=4"),
             height=700,
             type="messages",
-            show_label=True
+            show_label=True,
+            visible=False
         )
-
         
         with gr.Row(equal_height=True):
             topic = gr.Textbox(
                 label="Article Topic",
                 placeholder="Enter the topic you want an article about...",
-                #lines=2,
-                scale=4
+                scale=4,
+                visible=False
             )
             
-            async def process_input(topic, history):
+            async def process_input(topic, history, openai_api_key):
+                nonlocal article_crew
+                # Initialize ArticleCrew with the API key if not already initialized
+                if article_crew is None:
+                    article_crew = ArticleCrew(api_key=openai_api_key)
+                
                 history.append(ChatMessage(role="user", content=f"Write an article about: {topic}"))
                 yield history
                 
@@ -265,11 +288,25 @@ def create_demo():
                     history.extend(messages)
                     yield history
             
-            btn = gr.Button("Write Article", variant="primary", scale=1)
-            
+            btn = gr.Button("Write Article", variant="primary", scale=1, visible=False)
+
+        def show_interface():
+            return {
+                openai_api_key: gr.Textbox(visible=False),
+                chatbot: gr.Chatbot(visible=True),
+                topic: gr.Textbox(visible=True),
+                btn: gr.Button(visible=True)
+            }
+
+        openai_api_key.submit(
+            show_interface,
+            None,
+            [openai_api_key, chatbot, topic, btn]
+        )
+
         btn.click(
             process_input,
-            inputs=[topic, chatbot],
+            inputs=[topic, chatbot, openai_api_key],
             outputs=[chatbot]
         )
 
@@ -278,4 +315,4 @@ def create_demo():
 if __name__ == "__main__":
     demo = create_demo()
     demo.queue()
-    demo.launch(debug=True, share=True)
+    demo.launch()