Update app.py

app.py

@@ -12,35 +12,61 @@ load_dotenv()
 # Access the Groq API key
 groq_api_key = os.getenv("GROQ_API_KEY")
 
-agent = Agent(
-    model=Groq(id="llama-3.3-70b-versatile", api_key=groq_api_key),
-    tools=[DuckDuckGo(), Newspaper4k()],
-    description="You are a senior NYT researcher writing an article on a topic.",
-    instructions=[
-        "For a given topic, search for the top 5 links.",
-        "Then read each URL and extract the article text, if a URL isn't available, ignore it.",
-        "Analyse and prepare an NYT worthy article based on the information.",
-    ],
-    markdown=True,
-    show_tool_calls=True,
-    add_datetime_to_instructions=True,
-    # debug_mode=True,
-)
+# Initialize the agent
+try:
+    agent = Agent(
+        model=Groq(id="llama-3.3-70b-versatile", api_key=groq_api_key),
+        tools=[DuckDuckGo(), Newspaper4k()],
+        description="You are a senior NYT researcher writing an article on a topic.",
+        instructions=[
+            "For a given topic, search for the top 5 links.",
+            "Then read each URL and extract the article text, if a URL isn't available, ignore it.",
+            "Analyse and prepare an NYT-worthy article based on the information.",
+        ],
+        markdown=True,
+        show_tool_calls=True,
+        add_datetime_to_instructions=True,
+    )
+except Exception as e:
+    # Print error if the agent initialization fails
+    raise RuntimeError(f"Error initializing the agent: {e}")
 
+# Function to process input and generate an article
 def generate_article(topic):
-    response = agent.print_response(topic, stream=True)
-    # Collect the streamed responses
-    full_response = ""
-    for part in response:
-        full_response += part
-    return full_response
-
-iface = gr.Interface(
-    fn=generate_article,
-    inputs="text",
-    outputs="text",
-    title="NYT Article Generator",
-    description="Enter a topic to generate an NYT-style article.",
-)
-
-iface.launch()
+    if not topic.strip():
+        return "Please enter a valid topic."
+
+    try:
+        response = agent.run(topic)  # Use `run` for robustness
+        # Handle different response types
+        if isinstance(response, str):
+            return response  # Direct string output
+        elif isinstance(response, list):
+            return "\n".join(response)  # List of strings
+        elif hasattr(response, 'content'):  # Response object with `content` attribute
+            return response.content
+        else:
+            return f"Unexpected response type: {type(response)}. Raw output: {response}"
+    except Exception as e:
+        return f"Error generating the article: {e}"
+
+# Gradio interface
+with gr.Blocks() as app:
+    gr.Markdown("# 📰 NYT-Style Article Generator")
+    gr.Markdown(
+        "Enter a topic below, and the app will generate an NYT-style article by searching, extracting, and summarizing information from the web."
+    )
+
+    with gr.Row():
+        topic_input = gr.Textbox(
+            label="Enter Topic", placeholder="e.g., Simulation Theory", lines=1
+        )
+        generate_button = gr.Button("Generate Article")
+
+    output_text = gr.Markdown(label="Generated Article")
+
+    generate_button.click(fn=generate_article, inputs=topic_input, outputs=output_text)
+
+# Run the app
+if __name__ == "__main__":
+    app.launch()