Update app.py

app.py

@@ -1,73 +1,107 @@
 import gradio as gr
 import re
 import os
+import numpy as np
 from metrology_rag import MetrologyRAGSystem
 
-rag_system = None
+css = """
+footer {visibility: hidden}
+.markdown {max-width: 95% !important}
+"""
 
-def strip_ansi_codes(text):
-    return re.sub(r'\x1B\[[0-?]*[ -/]*[@-~]', '', text)
+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 initialize_system(files):
-    global rag_system
-    try:
-        temp_dir = "/kaggle/working/temp_pdfs/"
-        os.makedirs(temp_dir, exist_ok=True)
+    def process_query(self, query):
+        if not self.rag_system:
+            return "⚠️ Inicialize o sistema primeiro com documentos PDF!"
         
-        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())
+        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!"
         
-        rag_system = MetrologyRAGSystem()
-        rag_system.initialize_system(temp_dir)
-        return f"✅ Sistema inicializado com {len(rag_system.documents)} documentos!"
-    except Exception as e:
-        return f"❌ Erro: {str(e)}"
+        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 answer_query(query):
-    if not rag_system:
-        return "⚠️ Inicialize o sistema primeiro!"
-    try:
-        response = rag_system.generate_technical_response(query)
-        return strip_ansi_codes(response)
-    except Exception as e:
-        return f"Erro: {str(e)}"
+    def _clean_response(self, text):
+        return re.sub(r'\x1B\[[0-?]*[ -/]*[@-~]', '', text)
 
-def analyze_report(file):
-    if not rag_system:
-        return "⚠️ Sistema não inicializado!"
-    try:
-        temp_path = f"/kaggle/working/{os.path.basename(file.name)}"
-        with open(temp_path, "wb") as f:
-            f.write(file.read())
-        analysis = rag_system.analyze_metrology_report(temp_path)
-        return strip_ansi_codes(analysis)
-    except Exception as e:
-        return f"Erro: {str(e)}"
+interface = SpaceInterface()
 
-with gr.Blocks(title="Sistema Metrologia v2.1") as demo:
-    gr.Markdown("# 🔧 Sistema de Metrologia Inteligente")
+with gr.Blocks(css=css, title="Metrology Assistant") as demo:
+    gr.Markdown("# 🔧 Assistente de Metrologia Inteligente")
     
-    with gr.Tab("Inicialização"):
-        gr.Markdown("### Carregar Documentos PDF")
-        file_input = gr.File(file_count="multiple", file_types=[".pdf"])
-        init_btn = gr.Button("Inicializar")
-        init_status = gr.Textbox(label="Status")
-        init_btn.click(initialize_system, file_input, init_status)
-
-    with gr.Tab("Consulta"):
-        gr.Markdown("### Consulta Técnica")
-        query_input = gr.Textbox(label="Sua consulta")
-        query_btn = gr.Button("Enviar")
-        response_output = gr.Markdown(label="Resposta")
-        query_btn.click(answer_query, query_input, response_output)
+    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()
 
-    with gr.Tab("Análise"):
-        gr.Markdown("### Analisar Relatório")
-        report_input = gr.File(file_count="single", file_types=[".pdf"])
-        analyze_btn = gr.Button("Analisar")
-        analysis_output = gr.Markdown(label="Resultado")
-        analyze_btn.click(analyze_report, report_input, analysis_output)
+    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)