Spaces:
Runtime error
Runtime error
| from langchain.llms import GooglePalm | |
| from langchain.prompts import PromptTemplate | |
| import gradio as gr | |
| import pdfplumber | |
| api_key = 'AIzaSyDiDVR0LcoIakZvLXbDbA3W861X0ctDt38' # get this free api key from https://makersuite.google.com/ | |
| llm = GooglePalm(google_api_key=api_key, temperature=0.1) | |
| def extract_text(file_path): | |
| with open(file_path, 'rb') as f: | |
| pdf = pdfplumber.open(f) | |
| text = "" | |
| for page in pdf.pages: | |
| text += page.extract_text() | |
| prompt_template = PromptTemplate.from_template(""" | |
| **Q:** Make important notes and extract important information from the given below PDF text that is asked in any competitive exam. | |
| **PDF Text:** | |
| ``` | |
| {PDF_text} | |
| ``` | |
| **A:** | |
| ``` | |
| [Notes and important information] | |
| ```""" | |
| ) | |
| prompt = prompt_template.format(PDF_text=text) | |
| return llm(prompt) | |
| demo = gr.Interface( | |
| fn=extract_text, | |
| inputs="file", | |
| outputs="text", | |
| description="Extract Important inforamtion from PDF file", | |
| ) | |
| demo.launch(debug=True) |