Update app.py

app.py

@@ -1,11 +1,9 @@
 # app.py
 import gradio as gr
-from logic.generator import generate_code
+from logic.generator import generate_code, tokenizer, model
 from logic.fixer import fix_code
 from logic.refactor import refactor_code
 from logic.file_reader import read_uploaded_file
-
-from logic.generator import tokenizer, model
 import torch
 
 def process_file_command(arquivos, prompt):
@@ -58,23 +56,41 @@ with gr.Blocks(title="Python Code Assistant", css="#file-output textarea { overf
         btn_refac = gr.Button("Refatorar")
         btn_refac.click(fn=refactor_code, inputs=original_code, outputs=output_refac)
 
-    with gr.Tab("📂 Interpretar Arquivos"):
-        file_input = gr.File(label="Envie arquivos .py, .txt ou .zip", file_types=[".py", ".txt", ".zip"], file_count="multiple")
-        file_content = gr.Textbox(
-            label="Conteúdo interpretado",
-            lines=30,
+    with gr.Tab("📂 Interpretar e Comandar Arquivos"):
+        file_input = gr.File(
+            label="Arraste ou selecione arquivos (.py, .txt, .zip)",
+            file_types=[".py", ".txt", ".zip"],
+            file_count="multiple"
+        )
+
+        file_summary = gr.Textbox(
+            label="📄 Resumo dos arquivos enviados",
+            lines=15,
             max_lines=1000,
             interactive=False,
-            show_copy_button=True,
-            elem_id="file-output"
+            show_copy_button=True
+        )
+
+        command_file = gr.Textbox(
+            label="💬 O que deseja fazer com os arquivos?",
+            placeholder="Ex: Corrija todos e transforme em uma API com FastAPI"
         )
-        btn_file = gr.Button("Ler arquivos")
-        btn_file.click(fn=lambda files: "\n\n".join(read_uploaded_file(f) for f in files), inputs=file_input, outputs=file_content)
 
-    with gr.Tab("🤖 Comando com base nos arquivos"):
-        command_file = gr.Textbox(label="Diga o que quer que a IA faça com os arquivos enviados", placeholder="Ex: Corrija e aplique boas práticas")
-        command_output = gr.Code(label="Resultado")
+        command_output = gr.Code(label="⚙️ Resultado gerado pela IA")
         btn_command = gr.Button("Executar comando")
-        btn_command.click(fn=process_file_command, inputs=[file_input, command_file], outputs=command_output)
+
+        # Resumo automático dos arquivos
+        file_input.change(
+            fn=lambda files: "\n\n".join(read_uploaded_file(f)[:1000] for f in files),
+            inputs=file_input,
+            outputs=file_summary
+        )
+
+        # Executa comando com base nos arquivos carregados
+        btn_command.click(
+            fn=process_file_command,
+            inputs=[file_input, command_file],
+            outputs=command_output
+        )
 
 demo.launch()