Update app.py

app.py

@@ -207,11 +207,11 @@ def save_text_to_pdf(text, output_path):
    doc.save(output_path)
    print(f"Text saved to PDF at {output_path}.")
 # Function to handle user queries
-def handle_query(query, is_read_pdf, instructions):
+def handle_query(query, is_read_pdf, instructions, pdf_file=None):
    print("Handling user query...")
    max_chars_per_chunk = 1000  # Adjust this value as needed to control chunk size
-   if is_read_pdf:
-       pdf_text = read_pdf(query)
+   if is_read_pdf and pdf_file:
+       pdf_text = read_pdf(pdf_file)
        text_chunks = [pdf_text[i:i+max_chars_per_chunk] for i in range(0, len(pdf_text), max_chars_per_chunk)]
    else:
        search_results = google_search(query)
@@ -231,13 +231,14 @@ def handle_query(query, is_read_pdf, instructions):
 def run_app():
    with gr.Blocks() as demo:
        gr.Markdown("# Web and PDF Summarizer")
-       query = gr.Textbox(label="Enter your query or upload a PDF", placeholder="Enter query here")
+       query = gr.Textbox(label="Enter your query", placeholder="Enter query here")
+       pdf_file = gr.File(label="Upload PDF", file_types=[".pdf"])
        is_read_pdf = gr.Checkbox(label="Read PDF", value=False)
        instructions = gr.Textbox(label="Enter instructions", placeholder="Enter instructions here")
        output = gr.Textbox(label="Summary")
        clear_cache_btn = gr.Button("Clear Cache")
        clear_cache_btn.click(fn=clear_cache, outputs=output)
        generate_btn = gr.Button("Generate Summary")
-       generate_btn.click(fn=handle_query, inputs=[query, is_read_pdf, instructions], outputs=output)
+       generate_btn.click(fn=handle_query, inputs=[query, is_read_pdf, instructions, pdf_file], outputs=output)
    demo.launch()
 run_app()