big text

app.py

@@ -32,15 +32,14 @@ graph_config = {
 
 
 def scrape_and_summarize(prompt, source):
-    return source.name, prompt
-    # smart_scraper_graph = SmartScraperGraph(
-    #     prompt=prompt,
-    #     source=source,
-    #     config=graph_config
-    # )
-    # result = smart_scraper_graph.run()
-    # exec_info = smart_scraper_graph.get_execution_info()
-    # return result, prettify_exec_info(exec_info)
+    smart_scraper_graph = SmartScraperGraph(
+        prompt=prompt,
+        source=source,
+        config=graph_config
+    )
+    result = smart_scraper_graph.run()
+    exec_info = smart_scraper_graph.get_execution_info()
+    return result, prettify_exec_info(exec_info)
 
 
 # Gradio interface
@@ -54,8 +53,7 @@ with gr.Blocks() as demo:
         with gr.Column():
             model_dropdown = gr.Textbox(label="Model", value="Mistral-7B-Instruct-v0.2")
             prompt_input = gr.Textbox(label="Prompt", value="List me all the press releases with their headlines and urls.")
-            # source_input = gr.Textbox(label="Source URL", value="https://www.whitehouse.gov/")
-            file_input = gr.Interface(inputs=["file",], outputs="text")
+            source_input = gr.Textbox(label="HTML", value="html")
             scrape_button = gr.Button("Scrape and Summarize")
 
         with gr.Column():
@@ -64,7 +62,7 @@ with gr.Blocks() as demo:
 
     scrape_button.click(
         scrape_and_summarize,
-        inputs=[prompt_input, file_input],
+        inputs=[prompt_input, source_input],
         outputs=[result_output, exec_info_output]
     )