test

app.py

@@ -3,7 +3,6 @@ import pytesseract
 import os
 import pymupdf
 
-import streamlit as st
 import gradio as gr
 from prepare import prepare
 
@@ -61,12 +60,13 @@ def query_huggingface(text):
     return chat_model.invoke(text)
 
 # Gradio Interface for PDF Processing
-def process_file(file):
+def process_file(file, query):
     temp_file_path = "temp_uploaded_file"
     with open(temp_file_path, "wb") as temp_file:
         temp_file.write(file.read())
     pdf_output = read_pdf(temp_file_path)
-    return pdf_output
+    huggingface_output = query_huggingface(query)
+    return pdf_output, huggingface_output
 
 # Create Gradio App
 interface = gr.Interface(