Update app.py

app.py

@@ -202,17 +202,22 @@ with gr.Blocks(theme=gr.themes.Ocean()) as demo:
                 file_types=[".pdf"], file_count="multiple", label="Upload PDFs"
             )
 
+            gr.Markdown("## 2️⃣ Index the PDFs")
             convert_button = gr.Button("🔄 Index documents")
             message = gr.Textbox("Files not yet uploaded", label="Status")
-            api_key = gr.Textbox(
-                placeholder="Enter your Gemini API key here (must be valid)",
-                label="API key",
-            )
+            # api_key = gr.Textbox(
+            #     placeholder="Enter your Gemini API key here (must be valid)",
+            #     label="API key",
+            # )
             embeds = gr.State(value=[])
             imgs = gr.State(value=[])
 
         with gr.Column(scale=3):
-            gr.Markdown("## 2️⃣ Search")
+            gr.Markdown("## 3️⃣ Search")
+            api_key = gr.Textbox(
+                placeholder="Enter your Gemini API key here (must be valid)",
+                label="API key",
+            )
             query = gr.Textbox(placeholder="Enter your query here", label="Query")
             k = gr.Slider(
                 minimum=1,
@@ -225,11 +230,13 @@ with gr.Blocks(theme=gr.themes.Ocean()) as demo:
 
     # Define the output components
     search_button = gr.Button("🔍 Search", variant="primary")
-    gr.Markdown("## 1️⃣ Upload PDFs")
+
+    gr.Markdown("## 4️⃣ Retrieved Image")
     output_gallery = gr.Gallery(
         label="Retrieved Documents", height=600, show_label=True
     )
-    gr.Markdown("## 1️⃣ Upload PDFs")
+
+    gr.Markdown("## 5️⃣ Gemini Response")
     output_text = gr.Textbox(
         label="AI Response",
         placeholder="Generated response based on retrieved documents",