Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from huggingface_hub import hf_hub_download
 from llama_cpp import Llama
 
+# 1. Configuración del Modelo
 model_path = hf_hub_download(
     repo_id="bleondubos/CondorAI-7B", 
     filename="qwen2.5-coder-7b-instruct.Q4_K_M.gguf"
@@ -9,12 +10,9 @@ model_path = hf_hub_download(
 
 llm = Llama(model_path=model_path, n_ctx=4096, n_threads=4)
 
-# 2. Prediction Function (Backend)
+# 2. Función de Predicción
 def predict(message, history):
-    # System prompt para definir la personalidad de la IA
     system_prompt = "You are CondorAI, an expert AI in cybersecurity, SAST, and DAST. Provide technical, precise, and actionable security analysis."
-    
-    # Construcción del prompt ChatML con el historial de la conversación
     prompt = f"<|im_start|>system\n{system_prompt}<|im_end|>\n"
     
     for user_msg, assistant_msg in history:
@@ -36,40 +34,19 @@ def predict(message, history):
             partial_message += content
             yield partial_message
 
-with gr.Blocks() as demo:
-    gr.Markdown(
-        """
-        # 🦅 CondorAI Security Analyst
-        ### Real-time SAST/DAST code auditing and vulnerability analysis.
-        """
-    )
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="emerald", neutral_hue="slate")) as demo:
+    gr.Markdown("# 🦅 CondorAI Security Analyst")
+    gr.Markdown("### Real-time SAST/DAST code auditing and vulnerability analysis.")
     
-    chat = gr.ChatInterface(
+    gr.ChatInterface(
         fn=predict,
-        chatbot=gr.Chatbot(
-            show_label=False, 
-            container=True,
-            height=600
-        ),
-        textbox=gr.Textbox(
-            placeholder="Paste your code or ask a security question...", 
-            container=False, 
-            scale=7
-        ),
-        submit_btn="Analyze",
-        stop_btn="Stop",
-        retry_btn="Retry",
-        undo_btn="Undo",
-        clear_btn="Clear Chat",
+        type="messages",
+        chatbot=gr.Chatbot(height=600),
+        textbox=gr.Textbox(placeholder="Paste your code or ask a security question...", scale=7),
     )
     
-    gr.Markdown(
-        """
-        ---
-        **Note:** CondorAI uses Q4_K_M quantization to optimize performance on CPU. 
-        License: CC BY-NC-SA 4.0.
-        """
-    )
+    gr.Markdown("---")
+    gr.Markdown("**Note:** CondorAI uses Q4_K_M quantization. License: CC BY-NC-SA 4.0.")
 
 if __name__ == "__main__":
-    demo.launch(theme=gr.themes.Soft(primary_hue="emerald", neutral_hue="slate"))
+    demo.launch()