Spaces:

Chompi10
/

asddsad

Runtime error

App Files Files Community

Chompi10 commited on 3 days ago

Commit

b6a565f

verified ·

1 Parent(s): a918733

Update app.py

Browse files

Files changed (1) hide show

app.py +111 -102

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 app = Flask(__name__)
 MODEL_NAME = "microsoft/phi-2"
-MAX_NEW_TOKENS = 200
 print("Cargando modelo...")
@@ -23,141 +23,135 @@ model.eval()
 print("Modelo cargado correctamente.")
-# ---------- FRONTEND ----------
 @app.route("/")
 def index():
     return Response("""
 <!DOCTYPE html>
-<html lang="es">
 <head>
 <meta charset="UTF-8">
-<title>Asistente IA</title>
 <style>
 body {
-    margin: 0;
-    font-family: Arial, sans-serif;
-    background-color: #343541;
-    display: flex;
-    justify-content: center;
-    align-items: center;
-    height: 100vh;
 }
-.chat-container {
-    width: 700px;
-    height: 85vh;
-    background-color: #444654;
-    display: flex;
-    flex-direction: column;
-    border-radius: 10px;
-    overflow: hidden;
 }
-.chat-box {
-    flex: 1;
-    padding: 20px;
-    overflow-y: auto;
-    display: flex;
-    flex-direction: column;
 }
-.message {
-    margin-bottom: 15px;
-    padding: 12px;
-    border-radius: 8px;
-    max-width: 80%;
-    white-space: pre-wrap;
 }
-.user {
-    background-color: #19c37d;
-    align-self: flex-end;
-    color: black;
 }
-.bot {
-    background-color: #555869;
-    align-self: flex-start;
-    color: white;
 }
-.input-area {
-    display: flex;
-    border-top: 1px solid #555;
 }
-input {
-    flex: 1;
-    padding: 15px;
-    border: none;
-    outline: none;
-    font-size: 16px;
 }
-button {
-    padding: 15px;
-    border: none;
-    background-color: #19c37d;
-    cursor: pointer;
-    font-weight: bold;
 }
-button:hover {
-    opacity: 0.9;
 }
 </style>
 </head>
 <body>
-<div class="chat-container">
-    <div id="chat-box" class="chat-box"></div>
     <div class="input-area">
-        <input type="text" id="user-input" placeholder="Escribí tu mensaje..." />
-        <button onclick="sendMessage()">Enviar</button>
     </div>
 </div>
 <script>
-async function sendMessage() {
-    const input = document.getElementById("user-input");
-    const chatBox = document.getElementById("chat-box");
-    const userMessage = input.value.trim();
-    if (!userMessage) return;
-    addMessage(userMessage, "user");
-    input.value = "";
-    const loading = addMessage("Escribiendo...", "bot");
-    try {
-        const response = await fetch("/chat", {
-            method: "POST",
-            headers: { "Content-Type": "application/json" },
-            body: JSON.stringify({ message: userMessage })
         });
-        const data = await response.json();
         loading.remove();
-        addMessage(data.response || "Error en la respuesta", "bot");
-    } catch (error) {
         loading.remove();
-        addMessage("Error conectando con el servidor", "bot");
     }
 }
-function addMessage(text, sender) {
-    const chatBox = document.getElementById("chat-box");
-    const messageDiv = document.createElement("div");
-    messageDiv.classList.add("message", sender);
-    messageDiv.innerText = text;
-    chatBox.appendChild(messageDiv);
-    chatBox.scrollTop = chatBox.scrollHeight;
-    return messageDiv;
 }
-document.getElementById("user-input")
-    .addEventListener("keypress", function(e) {
-        if (e.key === "Enter") {
-            sendMessage();
-        }
-    });
 </script>
 </body>
@@ -165,7 +159,9 @@ document.getElementById("user-input")
 """, mimetype="text/html")
-# ---------- API ----------
 @app.route("/chat", methods=["POST"])
 def chat():
     try:
@@ -173,9 +169,22 @@ def chat():
         user_input = data.get("message", "")
         if not user_input:
-            return jsonify({"error": "Mensaje vacío"}), 400
-        prompt = f"User: {user_input}\nAssistant:"
         inputs = tokenizer(prompt, return_tensors="pt")
@@ -184,9 +193,9 @@ def chat():
                 **inputs,
                 max_new_tokens=MAX_NEW_TOKENS,
                 do_sample=True,
-                temperature=0.7,
-                top_p=0.9,
-                repetition_penalty=1.2,
                 no_repeat_ngram_size=3,
                 pad_token_id=tokenizer.eos_token_id
             )

 app = Flask(__name__)
 MODEL_NAME = "microsoft/phi-2"
+MAX_NEW_TOKENS = 400
 print("Cargando modelo...")
 print("Modelo cargado correctamente.")
+# ===============================
+# FRONTEND
+# ===============================
 @app.route("/")
 def index():
     return Response("""
 <!DOCTYPE html>
+<html>
 <head>
 <meta charset="UTF-8">
+<title>AI Assistant</title>
 <style>
 body {
+    margin:0;
+    font-family:Arial;
+    background:#343541;
+    display:flex;
+    justify-content:center;
+    align-items:center;
+    height:100vh;
 }
+.container{
+    width:800px;
+    height:90vh;
+    background:#444654;
+    display:flex;
+    flex-direction:column;
+    border-radius:10px;
 }
+.chat{
+    flex:1;
+    padding:20px;
+    overflow-y:auto;
+    display:flex;
+    flex-direction:column;
 }
+.msg{
+    padding:12px;
+    border-radius:8px;
+    margin-bottom:12px;
+    max-width:85%;
+    white-space:pre-wrap;
 }
+.user{
+    background:#19c37d;
+    align-self:flex-end;
+    color:black;
 }
+.bot{
+    background:#555869;
+    align-self:flex-start;
+    color:white;
 }
+.input-area{
+    display:flex;
+    border-top:1px solid #555;
 }
+input{
+    flex:1;
+    padding:15px;
+    border:none;
+    outline:none;
+    font-size:16px;
 }
+button{
+    padding:15px;
+    border:none;
+    background:#19c37d;
+    font-weight:bold;
+    cursor:pointer;
 }
+button:hover{
+    opacity:0.9;
 }
 </style>
 </head>
 <body>
+<div class="container">
+    <div id="chat" class="chat"></div>
     <div class="input-area">
+        <input id="input" placeholder="Write your message..." />
+        <button onclick="send()">Send</button>
     </div>
 </div>
 <script>
+async function send(){
+    const input = document.getElementById("input");
+    const chat = document.getElementById("chat");
+    const text = input.value.trim();
+    if(!text) return;
+    add(text,"user");
+    input.value="";
+    const loading = add("Thinking...","bot");
+    try{
+        const res = await fetch("/chat",{
+            method:"POST",
+            headers:{"Content-Type":"application/json"},
+            body:JSON.stringify({message:text})
         });
+        const data = await res.json();
         loading.remove();
+        add(data.response || "Error","bot");
+    }catch(e){
         loading.remove();
+        add("Server error","bot");
     }
 }
+function add(text,type){
+    const chat=document.getElementById("chat");
+    const div=document.createElement("div");
+    div.className="msg "+type;
+    div.innerText=text;
+    chat.appendChild(div);
+    chat.scrollTop=chat.scrollHeight;
+    return div;
 }
+document.getElementById("input").addEventListener("keypress",function(e){
+    if(e.key==="Enter"){ send(); }
+});
 </script>
 </body>
 """, mimetype="text/html")
+# ===============================
+# API CHAT
+# ===============================
 @app.route("/chat", methods=["POST"])
 def chat():
     try:
         user_input = data.get("message", "")
         if not user_input:
+            return jsonify({"error": "Empty message"}), 400
+        # PROMPT MULTI-LENGUAJE + CÓDIGO COMPLETO
+        system_prompt = """
+You are an advanced AI assistant.
+Always reply in the SAME language as the user.
+If the user asks for code:
+- Provide COMPLETE working code.
+- Include all imports.
+- Do not cut the code.
+- Use proper formatting.
+- Use markdown triple backticks for code blocks.
+Be clear and professional.
+"""
+        prompt = f"{system_prompt}\nUser: {user_input}\nAssistant:"
         inputs = tokenizer(prompt, return_tensors="pt")
                 **inputs,
                 max_new_tokens=MAX_NEW_TOKENS,
                 do_sample=True,
+                temperature=0.2,
+                top_p=0.7,
+                repetition_penalty=1.1,
                 no_repeat_ngram_size=3,
                 pad_token_id=tokenizer.eos_token_id
             )