Update app.py

app.py

@@ -80,21 +80,35 @@ def generate_plot(query, pdf_file):
     logging.info("Plot saved to file.")
     return temp_file.name
 
-def gradio_interface(pdf_file, query):
+def gradio_interface_o(pdf_file, query):
     logging.info("Gradio interface called.")
     plot_path = generate_plot(query, pdf_file)
     with open(plot_path, "r") as f:
         html_content = f.read()
     logging.info("Returning HTML content.")
     return html_content
+def gradio_interface(pdf_file, query):
+    logging.info("Gradio interface called.")
+    plot_path = generate_plot(query, pdf_file)
+    
+    # Assuming 'plot_path' can be accessed via a URL 'plot_url'
+    plot_url = f"http://yourserver.com/path/to/{os.path.basename(plot_path)}"
+    logging.info(f"Plot URL: {plot_url}")
+    
+    # Return an iframe pointing to the plot URL
+    iframe_html = f'<iframe src="{plot_url}" width="100%" height="700"></iframe>'
+    logging.info("Returning iframe HTML content.")
+    return iframe_html
 
+# Update the outputs to gr.HTML() if it's not already
 iface = gr.Interface(
     fn=gradio_interface,
     inputs=[gr.File(label="Upload PDF"), gr.Textbox(label="Query")],
-    outputs=gr.HTML(label="Visualization"),
+    outputs=gr.HTML(),
     title="PDF Content Visualizer",
     description="Upload a PDF and enter a query to visualize the content."
 )
 
+
 if __name__ == "__main__":
     iface.launch()