Update app.py

app.py

@@ -1,111 +1,70 @@
 import gradio as gr
-import re
 import os
+import re
 import numpy as np
 from metrology_rag import MetrologyRAGSystem
 
-css = """
-footer {visibility: hidden}
-.markdown {max-width: 95% !important}
-"""
-
-class SpaceInterface:
+class CalibrationAssistant:
     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)
-            
-            # Correção necessária aqui
-            for file in files:
-                file_path = os.path.join(temp_dir, os.path.basename(file.name))
-                # Usar open() no arquivo temporário
-                with open(file.name, "rb") as f:
-                    content = f.read()
-                    with open(file_path, "wb") as dest_file:
-                        dest_file.write(content)
-            
-            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!"
+        # Inicializa com documentos pré-carregados
+        self.rag_system = MetrologyRAGSystem()
+        self.rag_system.initialize_system("rag_documents")
         
+    def technical_query(self, query):
         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)}"
+            return f"Erro na consulta: {str(e)}"
 
-    def generate_report(self, report_title):
-        if not self.rag_system or not self.current_response:
-            return None, "⚠️ Execute uma consulta primeiro!"
-        
+    def analyze_report(self, file):
         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!"
+            analysis = self.rag_system.analyze_metrology_report(file.name)
+            return self._clean_response(analysis)
         except Exception as e:
-            return None, f"❌ Falha: {str(e)}"
+            return f"Erro na análise: {str(e)}"
 
     def _clean_response(self, text):
         return re.sub(r'\x1B\[[0-?]*[ -/]*[@-~]', '', text)
 
-interface = SpaceInterface()
+assistant = CalibrationAssistant()
 
-with gr.Blocks(css=css, title="Metrology Assistant") as demo:
-    gr.Markdown("# 🔧 Assistente de Metrologia Inteligente")
+css = """
+footer {visibility: hidden}
+.markdown {max-width: 95% !important}
+"""
+
+with gr.Blocks(css=css, title="Assistente de Metrologia") as demo:
+    gr.Markdown("# 🔧 Assistente de Metrologia Certificada")
     
-    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.Tab("Consulta Técnica"):
+        gr.Markdown("### Sistema RAG com Documentação Pré-Carregada")
+        query_input = gr.Textbox(
+            label="Faça sua consulta técnica",
+            placeholder="Ex: Qual o procedimento para calibração de micrômetros?..."
+        )
+        query_btn = gr.Button("Enviar")
+        response_output = gr.Markdown(label="Resposta Técnica Certificada")
         
-        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
-    )
+    with gr.Tab("Análise de Relatório"):
+        gr.Markdown("### Verificação de Conformidade de Relatórios")
+        report_input = gr.File(
+            label="Carregue seu relatório de calibração (PDF)",
+            file_types=[".pdf"]
+        )
+        analyze_btn = gr.Button("Analisar")
+        analysis_output = gr.Markdown(label="Resultado da Análise")
     
     query_btn.click(
-        interface.process_query,
+        assistant.technical_query,
         inputs=query_input,
         outputs=response_output
     )
     
-    report_btn.click(
-        interface.generate_report,
+    analyze_btn.click(
+        assistant.analyze_report,
         inputs=report_input,
-        outputs=[report_download, report_status]
+        outputs=analysis_output
     )
 
-demo.launch(debug=True)
+demo.launch()