Update app.py

app.py

@@ -34,7 +34,7 @@ processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-2B-Instruct", trust_rem
 @spaces.GPU()
 def process_pdf_and_query(pdf_file, user_query):
     # Convert the PDF to images
-    images = convert_from_path(pdf_file.name) 
+    images = convert_from_path(pdf_file.name)
     num_images = len(images)
 
     # Indexing the PDF in RAG
@@ -88,12 +88,26 @@ def process_pdf_and_query(pdf_file, user_query):
 
     return output_text[0], num_images
 
+
 pdf_input = gr.File(label="Upload PDF")  
 query_input = gr.Textbox(label="Enter your query", placeholder="Ask a question about the PDF") 
-output_text = gr.Textbox(label="Model Answer") 
+output_text = gr.Textbox(label="Model Answer")  
 output_images = gr.Textbox(label="Number of Images in PDF")  
 
 
+explanation = """
+<div style="text-align: center; margin-bottom: 20px;">
+    <h2 style="font-weight: bold; font-size: 24px;">Multimodal RAG (Retrieval-Augmented Generation)</h2>
+    <p>
+        This application utilizes the ColPali model as a multimodal retriever, 
+        which retrieves relevant information from documents and generates answers 
+        using the Qwen/Qwen2-VL-2B-Instruct LLM (Large Language Model) 
+        via the Byaldi library, developed by Answer.ai.
+    </p>
+</div>
+"""
+
+
 footer = """
 <div style="text-align: center; margin-top: 20px;">
     <a href="https://www.linkedin.com/in/pejman-ebrahimi-4a60151a7/" target="_blank">LinkedIn</a> |
@@ -107,18 +121,6 @@ footer = """
 </div>
 """
 
-explanation = """
-<div style="text-align: center; margin-bottom: 20px;">
-    <h2 style="font-weight: bold; font-size: 24px;">Multimodal RAG (Retrieval-Augmented Generation)</h2>
-    <p>
-        This application utilizes the ColPali model as a multimodal retriever, 
-        which retrieves relevant information from documents and generates answers 
-        using the Qwen/Qwen2-VL-2B-Instruct LLM (Large Language Model) 
-        via the Byaldi library, developed by Answer.ai.
-    </p>
-</div>
-"""
-
 
 demo = gr.Interface(
     fn=process_pdf_and_query, 
@@ -130,10 +132,12 @@ demo = gr.Interface(
 
 
 with demo:
-    gr.HTML(explanation) 
-    gr.HTML(footer)  
+    gr.HTML(explanation)  
+    gr.HTML(footer) 
     gr.DuplicateButton(value="Duplicate Space for private use", elem_classes="duplicate-button")  
-    gr.Button("Submit", elem_classes="submit-button")  
+
+    submit_btn = gr.Button("Submit", elem_classes="submit-button")  
+
 
 css = """
 <style>
@@ -145,6 +149,7 @@ css = """
         padding: 10px 20px;
         font-size: 16px;
         cursor: pointer;
+        margin: 10px; /* Add some space between buttons */
     }
     .duplicate-button {
         background-color: green;
@@ -160,4 +165,5 @@ css = """
 """
 gr.HTML(css)
 
-demo.launch(debug=True) 
+# Launch the Gradio app
+demo.launch(debug=True)