Update app.py

app.py

@@ -43,12 +43,10 @@ def process_pdf_and_query(pdf_file, user_query):
         overwrite=True
     )
 
-    # Search the query in the RAG model
     results = RAG.search(user_query, k=1)
     if not results:
         return "No results found.", num_images
 
-    # Retrieve the page number and process image
     image_index = results[0]["page_num"] - 1
     messages = [
         {
@@ -63,7 +61,6 @@ def process_pdf_and_query(pdf_file, user_query):
         }
     ]
 
-
     text = processor.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
     image_inputs, video_inputs = process_vision_info(messages)
     inputs = processor(
@@ -85,7 +82,6 @@ def process_pdf_and_query(pdf_file, user_query):
 
     return output_text[0], num_images
 
-
 css = """
     .duplicate-button {
         background-color: #6272a4;
@@ -104,6 +100,12 @@ css = """
         border-radius: 10px;
         box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
     }
+    .title {
+        font-size: 30px;
+        font-weight: bold;
+        text-align: center;
+        margin-bottom: 20px;
+    }
 """
 
 explanation = """
@@ -126,18 +128,18 @@ footer = """
 </div>
 """
 
-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_images = gr.Textbox(label="Number of Images in PDF") 
+pdf_input = gr.File(label="Upload PDF")  # Single PDF file input
+query_input = gr.Textbox(label="Enter your query", placeholder="Ask a question about the PDF")  # User query input
+output_text = gr.Textbox(label="Model Answer")  # Output for the model's answer
+output_images = gr.Textbox(label="Number of Images in PDF")  # Output for number of images
 duplicate_button = gr.DuplicateButton(value="Duplicate Space for private use", elem_classes="duplicate-button")
 
 # Launch the Gradio app 
 demo = gr.Interface(
     fn=process_pdf_and_query, 
-    inputs=[pdf_input, query_input],  
-    outputs=[output_text, output_images],  
-    title="Multimodal RAG with Image Query - By Pejman Ebrahimi - Please like the space if it is useful",
+    inputs=[pdf_input, query_input],  # List of inputs
+    outputs=[output_text, output_images],  # List of outputs
+    title="",
     theme='freddyaboulton/dracula_revamped',
     css=css,
     description=explanation,
@@ -145,7 +147,13 @@ demo = gr.Interface(
 )
 
 with demo:
+    gr.HTML("""
+    <div class='title'>
+        Multimodal RAG with Image Query - 
+        <a href="https://github.com/arad1367" target="_blank" style="color: #ff79c6; text-decoration: none;">
+            Pejman Ebrahimi
+        </a>
+    </div>
+    """)
     gr.HTML(footer)
     duplicate_button
-
-demo.launch(debug=True)