Update app.py

app.py

@@ -52,12 +52,12 @@ def create_prompt(top_k_list: list[dict], question: str) -> str:
 
 """Gradio Application"""
 
-def process_files(ground_truth_file, pdf_files):
+ def process_files(ground_truth_file, pdf_files):
     # Process ground truth file
     if ground_truth_file.name.endswith('.csv'):
-        queries_t = pxt.io.import_csv(rag_demo.queries, ground_truth_file.name)
+        queries_t = pxt.io.import_csv('rag_demo.queries', ground_truth_file.name)
     else:
-        queries_t = pxt.io.import_excel(rag_demo.queries, ground_truth_file.name)
+        queries_t = pxt.io.import_excel('rag_demo.queries', ground_truth_file.name)
 
     # Process PDF files
     documents_t = pxt.create_table(
@@ -110,11 +110,7 @@ def process_files(ground_truth_file, pdf_files):
         }
     ]
 
-def query_llm(question, ground_truth_file, pdf_files):
-    queries_t = pxt.get_table('rag_demo.queries')
-    chunks_t = pxt.get_table('rag_demo.chunks')
-
-
+def query_llm(question):
 
     # Add OpenAI response column
     queries_t['response'] = openai.chat_completions(
@@ -173,5 +169,4 @@ with gr.Blocks() as demo:
     query_button.click(query_llm, inputs=question_input, outputs=output_dataframe)
 
 if __name__ == "__main__":
-    demo.launch()
-
+    demo.launch()