Upload gradio_app.py

gradio_app.py

@@ -0,0 +1,50 @@
+import gradio as gr
+import requests
+
+BASE_URL = "https://citegraph-1099254244451.us-east1.run.app/"
+HEADERS = {"Content-Type": "application/json"}
+
+
+def load_documents(file_path):
+    print(file_path)
+    url = f"{BASE_URL}/api/predict"
+    if file_path is not None:
+        with open(file_path.name, "rb") as f:
+            file = {"file": f}
+            response = requests.post(url, files=file)
+        f.close()
+
+    output = response.json()
+    print(output)
+    return f"## Output: \n# <div align='center'>{output['predicted_label']}</div>"
+
+
+def createUI():
+    css = """
+        h1 {
+            text-align: center;
+            display:block;
+        }
+    """
+    with gr.Blocks(css=css) as demo:
+        with gr.Row():
+            with gr.Column(scale=1):
+                # pdf_display = gr.Image(
+                #     label="Uploaded PDF Page", interactive=False, height=680
+                # )
+                upload_btn = gr.File(label="Upload a PDF", file_types=[".pdf"])
+
+        with gr.Row():
+            class_label = gr.Markdown("## Output: ")
+
+        upload_btn.upload(
+            fn=load_documents,
+            inputs=[upload_btn],
+            outputs=[class_label],
+        )
+    return demo
+
+
+if __name__ == "__main__":
+    demo = createUI()
+    demo.launch(pwa=True, share=False)