Spaces:
Running
Running
File size: 2,073 Bytes
fc5e5f2 3117482 897be90 822d7ea 7fab762 822d7ea 90592bd 822d7ea c4b5495 b138465 822d7ea 3117482 8ccf021 a5554cc 2eb9ec0 a5554cc 8ccf021 c4b5495 2eb9ec0 c4b5495 8ccf021 d6fee21 3117482 154bdd5 b1c9942 641a65d 3117482 c07956a 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 80 |
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():
textoutput = gr.Markdown(label="Antwort", elem_id="md", value="#")
with gr.Row():
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, textoutput])
demo.launch() |