Main

app.py

@@ -36,7 +36,6 @@ def summarize(slides):
 
 
 def convert2markdown(generate_slides):
-    # save_path = f"tmp/{FILENAME}"
     mdFile = MdUtils(file_name=FILENAME, title=f'{FILENAME} Presentation')
     for k, v in generate_slides.items():
         mdFile.new_paragraph('---')
@@ -54,6 +53,7 @@ def convert2markdown(generate_slides):
 
 def inference(document):
     global FILENAME
+    print(document)
     doc = fitz.open(document)
     FILENAME = Path(doc.name).stem
     font_counts, styles = preprocess.get_font_info(doc, granularity=False)
@@ -62,14 +62,15 @@ def inference(document):
     slides = preprocess.get_slides(texts)
     generated_slides = summarize(slides)
     markdown_path = convert2markdown(generated_slides)
-    with open(markdown_path, 'rt') as f:
-        markdown_str = f.read()
-    return markdown_str
+    # with open(markdown_path, 'rt') as f:
+    #     markdown_str = f.read()
+    return markdown_path
 
 
 with gr.Blocks() as demo:
-    inp = gr.File( file_types=['pdf'])
-    out = gr.Textbox(label="Markdown Content")
+    inp = gr.File(file_types=['pdf'])
+    out = gr.File(label="Markdown File")
+    # out = gr.Textbox(label="Markdown Content")
     inference_btn = gr.Button("Summarized PDF")
     inference_btn.click(fn=inference, inputs=inp, outputs=out, show_progress=True, api_name="summarize")