Spaces:
Sleeping
Sleeping
File size: 3,557 Bytes
853a257 26610a8 853a257 26610a8 853a257 26610a8 853a257 26610a8 853a257 26610a8 853a257 26610a8 853a257 26610a8 853a257 26610a8 853a257 26610a8 853a257 26610a8 853a257 26610a8 853a257 |
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 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 |
import gradio as gr
import re
import os
import numpy as np
from metrology_rag import MetrologyRAGSystem
css = """
footer {visibility: hidden}
.markdown {max-width: 95% !important}
"""
class SpaceInterface:
def __init__(self):
self.rag_system = None
self.current_response = ""
def initialize_system(self, files):
try:
temp_dir = "/tmp/metrology_pdfs"
os.makedirs(temp_dir, exist_ok=True)
for file in files:
file_path = os.path.join(temp_dir, os.path.basename(file.name))
with open(file_path, "wb") as f:
f.write(file.read())
self.rag_system = MetrologyRAGSystem()
self.rag_system.initialize_system(temp_dir)
return f"✅ Sistema inicializado com {len(self.rag_system.documents)} documentos!"
except Exception as e:
return f"❌ Erro: {str(e)}"
def process_query(self, query):
if not self.rag_system:
return "⚠️ Inicialize o sistema primeiro com documentos PDF!"
try:
response = self.rag_system.generate_technical_response(query)
self.current_response = response
return self._clean_response(response)
except Exception as e:
return f"Erro: {str(e)}"
def generate_report(self, report_title):
if not self.rag_system or not self.current_response:
return None, "⚠️ Execute uma consulta primeiro!"
try:
report_path = "/tmp/relatorio_tecnico.md"
self.rag_system.generate_report(report_title, self.current_response, report_path)
return report_path, "✅ Relatório gerado com sucesso!"
except Exception as e:
return None, f"❌ Falha: {str(e)}"
def _clean_response(self, text):
return re.sub(r'\x1B\[[0-?]*[ -/]*[@-~]', '', text)
interface = SpaceInterface()
with gr.Blocks(css=css, title="Metrology Assistant") as demo:
gr.Markdown("# 🔧 Assistente de Metrologia Inteligente")
with gr.Row():
with gr.Column(scale=2):
file_input = gr.File(
file_count="multiple",
file_types=[".pdf"],
label="Upload de Documentos Técnicos"
)
init_btn = gr.Button("Inicializar Sistema")
init_status = gr.Textbox(label="Status", interactive=False)
with gr.Column(scale=3):
query_input = gr.Textbox(
label="Consulta Técnica",
placeholder="Digite sua pergunta sobre metrologia..."
)
query_btn = gr.Button("Enviar Consulta")
response_output = gr.Markdown(label="Resposta Técnica")
report_input = gr.Textbox(
label="Título do Relatório",
placeholder="Digite o título do relatório..."
)
report_btn = gr.Button("Gerar Relatório")
report_download = gr.File(label="Download do Relatório")
report_status = gr.Markdown()
init_btn.click(
interface.initialize_system,
inputs=file_input,
outputs=init_status
)
query_btn.click(
interface.process_query,
inputs=query_input,
outputs=response_output
)
report_btn.click(
interface.generate_report,
inputs=report_input,
outputs=[report_download, report_status]
)
demo.launch(debug=True) |