Spaces:
Running
Running
File size: 2,042 Bytes
fc5e5f2 3117482 897be90 822d7ea 7fab762 822d7ea 90592bd 822d7ea c4b5495 b138465 822d7ea 3117482 8ccf021 a5554cc 947850d a5554cc 8ccf021 c4b5495 25b7d79 c4b5495 8ccf021 d6fee21 3117482 90592bd 641a65d 3117482 e12f54f 3117482 d6fee21 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 |
import gradio as gr
from PyPDF2 import PdfReader
import os
from docx import Document as DocxDocument
from gradio_client import Client
from IPython.display import display
from IPython.display import Markdown
custom_css = """
#md {
height: 450px;
font-size: 35px;
background: black;
padding: 20px;
padding-top: 40px;
color: white;
border: 1 px solid #383838;
}
"""
def routing(message):
client = Client("mgokg/SemanticRouting")
result = client.predict(
prompt=f"{message}",
api_name="/predict"
)
return result
def mitteilung(input):
client = Client("mgokg/Gemini2.0")
result = client.predict(
input_text=f"{input}",
api_name="/generate_press_release"
)
return result
def process_pdf(file):
# Read the PDF content
pdf_reader = PdfReader(file.name)
text = ""
for page in pdf_reader.pages:
text += page.extract_text()
return text
def process_file(file):
file_extension = file.name.split(".")[-1].lower()
if file_extension == 'pdf':
ocr_text = process_pdf(file)
result = mitteilung(ocr_text)
#display(Markdown(result))
return result
return ocr_text
elif file_extension == 'docx':
docx_document = DocxDocument(file.name)
text = ""
for paragraph in docx_document.paragraphs:
text += paragraph.text + "\n"
result = mitteilung(text)
display(Markdown(result))
return result
return text
#return [Document(text=text)]
with gr.Blocks() as demo:
gr.Markdown("### File upload", elem_classes="tab-header")
with gr.Row():
text_output = gr.Markdown(label="Antwort", elem_id="md", value="# PR")
#text_output = gr.Textbox(label="output")
with gr.Row():
file_input = gr.File(label="Wähle eine PDF-Datei aus", type="filepath")
with gr.Row():
submit_button = gr.Button("upload")
submit_button.click(process_file, inputs=file_input, outputs=text_output)
demo.launch() |