Update app.py

app.py

@@ -5,7 +5,7 @@ from crew import get_crew
 
 LLM = "gpt-4o"
 
-def invoke(openai_api_key, topic):
+def invoke(openai_api_key, topic, length):
     if (openai_api_key == ""):
         raise gr.Error("OpenAI API Key is required.")
     if (topic == ""):
@@ -15,13 +15,14 @@ def invoke(openai_api_key, topic):
 
     os.environ["OPENAI_API_KEY"] = openai_api_key
     
-    return get_crew(LLM).kickoff(inputs={"topic": topic})
+    return get_crew(LLM).kickoff(inputs={"topic": topic, "length": length})
 
 gr.close_all()
 
 demo = gr.Interface(fn = invoke, 
                     inputs = [gr.Textbox(label = "OpenAI API Key", type = "password", lines = 1),
-                              gr.Textbox(label = "Topic", value="Evolution of Retrieval-Augmented Generation from Naive RAG to Agentic RAG", lines = 1)],
+                              gr.Textbox(label = "Topic", value="Evolution of Retrieval-Augmented Generation from Naive RAG to Agentic RAG", lines = 1),
+                              gr.Number(label = "Words", value="2500")],
                     outputs = [gr.Textbox(label = "Output", lines = 1)],
                     title = "Multi-Agent RAG: Blog Post Generation",
                     description = os.environ["DESCRIPTION"])