Update app.py

app.py

@@ -84,24 +84,23 @@ def retrieve_info(query):
         
     return {"results": formatted_results}
 
-# Gradio interface
+def combined_interface(query, file, collection_name):
+    if query:
+        return retrieve_info(query)
+    elif file is not None and collection_name:
+        return embed_pdf(file, collection_name)
+    else:
+        return "Please enter a query or upload a PDF file."
+
 iface = gr.Interface(
-    fn=retrieve_info,
+    fn=combined_interface,
     inputs=[
-        gr.Textbox(label="Query")
+        gr.Textbox(label="Query", optional=True),
+        gr.File(label="PDF File", type="file", optional=True),
+        gr.Textbox(label="Collection Name", optional=True)
     ],
     outputs="text",
     allow_flagging="never"
 )
 
-# Embed PDF function
-iface.add_endpoint(
-    fn=embed_pdf,
-    inputs=[
-        gr.File(label="PDF File", type="file"),
-        gr.Textbox(label="Collection Name")
-    ],
-    outputs="text"
-)
-
 iface.launch()