# -*- coding: utf-8 -*- """app.ipynb Automatically generated by Colab. Original file is located at https://colab.research.google.com/drive/1XblbxoRxB4XOHixjGij789FPD9KjKdhi """ import os import pdfplumber import gradio as gr from langchain_groq.chat_models import ChatGroq # Set Groq API key securely GROQ_API_KEY = os.getenv("GROQ_API_KEY") # Fetch from environment variables if not GROQ_API_KEY: raise ValueError("GROQ_API_KEY is not set. Add it in Hugging Face Secrets.") # Initialize LLM (Mistral-8x7B) llm = ChatGroq(model_name="mixtral-8x7b-32768") def extract_text_from_pdf(pdf_file): """Extract text from a PDF file using pdfplumber.""" text = "" with pdfplumber.open(pdf_file) as pdf: for page in pdf.pages: text += page.extract_text() or "" # Extract text safely return text.strip() def summarize_text(text, length, style): """Summarize text with adjustable length and style.""" prompt = ( f""" Read the following document and summarize it in {style.lower()} format. Keep the summary {length.lower()}. Follow this step-by-step approach: 1. Identify key sections. 2. Extract essential points. 3. Remove unnecessary details. 4. Ensure factual accuracy without adding extra information. Document: {text[:10000]} # Limit input to 10,000 characters for efficiency """ ) response = llm.predict(prompt) return response.strip() def process_pdf(file, length, style): """Extract text and summarize PDF using Mistral-8x7B with customization.""" if not file: return "No file uploaded." text = extract_text_from_pdf(file.name) # Extract text if not text: return "Could not extract text from the PDF. Try another file." return summarize_text(text, length, style) # Create Gradio Interface interface = gr.Interface( fn=process_pdf, inputs=[ gr.File(label="Upload a PDF"), gr.Radio(["Short", "Medium", "Long"], label="Summary Length", value="Medium"), gr.Radio(["Bullets", "Key Takeaways", "Concise Paragraph"], label="Summary Style", value="Key Takeaways"), ], outputs="text", title="📄 Advanced PDF Summarizer", description="Upload a PDF file and get a structured summary with customization options." ) # Run the app interface.launch()