Update app.py

app.py

@@ -110,10 +110,10 @@ def format_relevant_documents(relevant_docs: list):
 
 
 # Function to handle PDF processing API call
-def process_pdf(pdf_file):
+def process_pdf(pdf_file, client_name):
     return client.predict(
         pdf_file=handle_file(pdf_file),
-        client_name="rosariarossi",  # Hardcoded client name
+        client_name=client_name,  # Hardcoded client name
         api_name="/process_pdf2"
     )[1]  # Return only the result string
 
@@ -211,14 +211,26 @@ with gr.Blocks(css=CSS) as demo:
             ],
         )
 
+    # with gr.Tab("Process PDF"):
+    #     pdf_input = gr.File(label="Upload PDF File")
+    #     pdf_output = gr.Textbox(label="PDF Result", interactive=False)
+
+    #     pdf_button = gr.Button("Process PDF")
+    #     pdf_button.click(
+    #         process_pdf,
+    #         inputs=[pdf_input],
+    #         outputs=pdf_output
+    #     )
+
     with gr.Tab("Process PDF"):
         pdf_input = gr.File(label="Upload PDF File")
+        select_client_dropdown = gr.Dropdown(['rosariarossi', 'bianchifiordaliso', 'lorenzoverdi'], value="rosariarossi", label="Select Client")
         pdf_output = gr.Textbox(label="PDF Result", interactive=False)
-
+    
         pdf_button = gr.Button("Process PDF")
         pdf_button.click(
             process_pdf,
-            inputs=[pdf_input],
+            inputs=[pdf_input, select_client_dropdown],  # Pass both PDF and client name as inputs
             outputs=pdf_output
         )