Create App.py

App.py

@@ -0,0 +1,73 @@
+import gradio as gr
+import re
+import os
+from metrology_rag import MetrologyRAGSystem
+
+rag_system = None
+
+def strip_ansi_codes(text):
+    return re.sub(r'\x1B\[[0-?]*[ -/]*[@-~]', '', text)
+
+def initialize_system(files):
+    global rag_system
+    try:
+        temp_dir = "/kaggle/working/temp_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())
+        
+        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)}"
+
+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 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)}"
+
+with gr.Blocks(title="Sistema Metrologia v2.1") as demo:
+    gr.Markdown("# 🔧 Sistema 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.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)
+
+demo.launch(debug=True)