Update app.py

app.py

@@ -0,0 +1,49 @@
+import fitz
+from transformers import GPT2Tokenizer, GPT2Model
+import gradio as gr
+
+def extract_text_from_pdf(pdf_path):
+    doc = fitz.open(pdf_path)
+    text = ""
+    for page_num in range(doc.page_count):
+        page = doc[page_num]
+        text += page.get_text()
+    return text
+
+def generate_review(summary):
+    # Your custom logic to generate a review
+    review = "The book conveys a powerful message about..."
+    return review
+
+def generate_summary(pdf_path):
+    # Load pre-trained GPT-2 model and tokenizer
+    tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
+    model = GPT2Model.from_pretrained('gpt2')
+
+    # Extract text from the PDF
+    input_text = extract_text_from_pdf(pdf_path)
+
+    # Tokenize and generate summary
+    input_tokens = tokenizer.encode(input_text, return_tensors="pt")
+    summary = model.generate(input_tokens)
+
+    # Return results
+    return {
+        "Extracted Information": input_text,
+        "Book Summary": tokenizer.decode(summary[0], skip_special_tokens=True),
+        "Review": generate_review(summary)
+    }
+
+iface = gr.Interface(
+    fn=generate_summary,
+    inputs=gr.File(),
+    outputs=[
+        gr.Textbox(),
+        gr.Textbox(),
+        gr.Textbox(),
+    ],
+    live=True,
+    capture_session=True
+)
+
+iface.launch()