Update app.py

app.py

@@ -8,8 +8,8 @@ os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 # Function to list uploaded PDFs
 def list_pdfs():
     files = os.listdir(UPLOAD_FOLDER)
-    pdf_list = [f'<a href="/file={UPLOAD_FOLDER}/{file}" target="_blank">{file}</a>' for file in files]
-    return '\n'.join(pdf_list)
+    pdf_list = [f'<a href="javascript:view_pdf(\'{file}\')">{file}</a>' for file in files]
+    return '<br>'.join(pdf_list)
 
 # Function to handle PDF upload
 def upload_pdf(pdf_file):
@@ -26,8 +26,20 @@ with gr.Blocks() as demo:
         pdf_list = gr.HTML(list_pdfs)
         pdf_upload = gr.File(label="Upload PDF", file_types=[".pdf"])
         upload_btn = gr.Button("Upload")
+        pdf_viewer = gr.HTML("<iframe id='pdf_viewer' style='width:100%; height:500px;'></iframe>")
 
     upload_btn.click(upload_pdf, inputs=pdf_upload, outputs=pdf_list)
 
+    # Custom JS to view PDF in iframe
+    js = """
+    <script>
+        function view_pdf(file) {
+            const viewer = document.getElementById('pdf_viewer');
+            viewer.src = `/file=uploads/${file}`;
+        }
+    </script>
+    """
+    pdf_list.update(js)
+
 # Run the app
 demo.launch()