Update app.py

app.py

@@ -4,18 +4,16 @@ import os
 import spaces
 
 @spaces.GPU
-def query_engine(llm_model, documents):
+def query_engine(llm_model, document):
     api_key = os.getenv("HUGGINGFACE_API_KEY")
     llm_api_base = "https://api-inference.huggingface.co/models/"
 
-    # Extract text content from uploaded files
-    docs = []
-    for doc in documents:
-        content = doc.decode("utf-8")
-        docs.append(content)
+
+    content = document.value.decode("utf-8")
+    docs.append(content)
 
     query_engine = AutoQueryEngine.from_defaults(
-        documents=docs,
+        documents=document,
         llm_model=llm_model,
         llm_api_base=llm_api_base,
     )
@@ -25,9 +23,9 @@ interface = gr.Blocks()
 with interface:
     gr.Markdown("# AutoQueryEngine Interface")
     llm_model = gr.Textbox(label="LLM Model", value="mistralai/Mixtral-8x7B-Instruct-v0.1")
-    documents = gr.File(label="Documents", file_count="multiple")
+    document = gr.File(label="Documents")
     query = gr.Textbox(label="Query")
     output = gr.Textbox(label="Output")
     query_btn = gr.Button("Query")
-    query_btn.click(fn=query_engine, inputs=[llm_model, documents], outputs=output)
+    query_btn.click(fn=query_engine, inputs=[llm_model, document], outputs=output)
 interface.launch()