Reorganize

Dockerfile

@@ -15,5 +15,3 @@ WORKDIR $HOME/app
 COPY --chown=user . $HOME/app
 
 ENTRYPOINT python app.py
-
-# sbt "runMain org.allenai.pdffigures2.FigureExtractorBatchCli 2304.11968v1.Track_Anything_Segment_Anything_Meets_Videos.pdf -m figures -t 48 -q"

app.py

@@ -29,12 +29,20 @@ def extract_figure(url):
     # return first figure
     return figure_path
 
+
+def run():
+
 # define input and output interfaces
-inputs = gr.inputs.Textbox(label="Enter URL of PDF file:")
-outputs = gr.outputs.Image(label="First figure in PDF:", type="filepath")
+    inputs = gr.inputs.Textbox(label="Enter URL of PDF file:")
+    outputs = gr.outputs.Image(label="First figure in PDF:", type="filepath")
+
+    # create interface
+    interface = gr.Interface(fn=extract_figure, inputs=inputs, outputs=outputs, title="Extract First Figure from PDF", description="Enter the URL of a PDF file and the first figure in the file will be extracted and displayed.")
+
+    # launch interface
+    interface.launch(server_name="0.0.0.0", server_port=7860)
+
 
-# create interface
-interface = gr.Interface(fn=extract_figure, inputs=inputs, outputs=outputs, title="Extract First Figure from PDF", description="Enter the URL of a PDF file and the first figure in the file will be extracted and displayed.")
+if __name__ == "__main__":
 
-# launch interface
-interface.launch()
+    run()