Spaces:

Chompi10
/

asddsad

Runtime error

App Files Files Community

Chompi10 commited on 16 days ago

Commit

3c2f1c9

verified ·

1 Parent(s): 6929287

Update app.py

Browse files

Files changed (1) hide show

app.py +112 -152

app.py CHANGED Viewed

@@ -3,13 +3,19 @@ import torch
 from flask import Flask, request, jsonify, Response
 from transformers import AutoTokenizer, AutoModelForCausalLM
 app = Flask(__name__)
 MODEL_NAME = "microsoft/phi-2"
-MAX_NEW_TOKENS = 400
 print("Cargando modelo...")
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForCausalLM.from_pretrained(
@@ -24,8 +30,75 @@ print("Modelo cargado correctamente.")
 # ===============================
-# FRONTEND
 # ===============================
 @app.route("/")
 def index():
     return Response("""
@@ -33,70 +106,20 @@ def index():
 <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>
@@ -104,6 +127,11 @@ button:hover{
 <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>
@@ -113,6 +141,7 @@ button:hover{
 async function send(){
     const input = document.getElementById("input");
     const chat = document.getElementById("chat");
     const text = input.value.trim();
     if(!text) return;
@@ -126,7 +155,7 @@ async function send(){
         const res = await fetch("/chat",{
             method:"POST",
             headers:{"Content-Type":"application/json"},
-            body:JSON.stringify({message:text})
         });
         const data = await res.json();
@@ -162,97 +191,24 @@ document.getElementById("input").addEventListener("keypress",function(e){
 # ===============================
 # API CHAT
 # ===============================
 @app.route("/chat", methods=["POST"])
 def chat():
     try:
         data = request.get_json()
         user_input = data.get("message", "")
         if not user_input:
             return jsonify({"error": "Empty message"}), 400
-        # PROMPT MULTI-LENGUAJE + CÓDIGO COMPLETO
-        system_prompt = """
-        GENERATOR_PROMPT = """
-You are a Senior Software Engineer.
-Generate:
-- Clean
-- Secure
-- Optimized
-- Production-ready code
-Explain briefly what the code does.
-Include best practices.
-"""
-ULTRA_REVIEW_PROMPT = """
-You are a Principal Software Architect and Security Engineer.
-Perform an EXTREMELY CRITICAL and PROFESSIONAL code review.
-Be strict. Be precise.
-Analyze:
-1. Logical correctness
-2. Edge cases
-3. Security vulnerabilities (OWASP mindset)
-4. Input validation
-5. Authentication flaws
-6. Concurrency risks
-7. Memory inefficiencies
-8. Performance bottlenecks
-9. Scalability risks
-10. SOLID violations
-11. Clean Architecture violations
-12. Code smells
-13. Dependency risks
-14. Error handling weaknesses
-15. Logging gaps
-16. Deployment risks
-Return STRICTLY in this structure:
-## Executive Summary
-## Critical Bugs
-## Security Vulnerabilities
-## Architecture Problems
-## Scalability Risks
-## Performance Issues
-## Code Quality Problems
-## Missing Edge Cases
-## Refactored Version
-## Final Verdict
-"""
-ARCHITECTURE_ANALYSIS_PROMPT = """
-You are a Senior Software Architect.
-Analyze the provided system from an ARCHITECTURAL perspective.
-Evaluate:
-- Layered architecture
-- Coupling and cohesion
-- Domain boundaries
-- API design
-- Data flow
-- Scalability model
-- Fault tolerance
-- Observability
-- Cloud readiness
-- Caching strategy
-- Database strategy
-- Horizontal scaling capability
-Return response in this format:
-## Architecture Overview
-## Structural Problems
-## Scalability Evaluation
-## Resilience Evaluation
-## Design Improvements
-## Suggested Architecture
-## Final Architectural Verdict
-"""
         prompt = f"{system_prompt}\nUser: {user_input}\nAssistant:"
@@ -263,10 +219,10 @@ Return response in this format:
                 **inputs,
                 max_new_tokens=MAX_NEW_TOKENS,
                 do_sample=True,
-                temperature=0.8,
-                top_p=0.8,
-                repetition_penalty=1.1,
-                no_repeat_ngram_size=2,
                 pad_token_id=tokenizer.eos_token_id
             )
@@ -279,6 +235,10 @@ Return response in this format:
         return jsonify({"error": str(e)}), 500
 if __name__ == "__main__":
     port = int(os.environ.get("PORT", 7860))
     app.run(host="0.0.0.0", port=port)

 from flask import Flask, request, jsonify, Response
 from transformers import AutoTokenizer, AutoModelForCausalLM
+# ===============================
+# CONFIGURACIÓN BÁSICA
+# ===============================
 app = Flask(__name__)
 MODEL_NAME = "microsoft/phi-2"
+MAX_NEW_TOKENS = 250
 print("Cargando modelo...")
+torch.set_num_threads(4)
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForCausalLM.from_pretrained(
 # ===============================
+# PROMPTS PROFESIONALES
 # ===============================
+GENERATOR_PROMPT = """
+You are a Senior Software Engineer.
+Always respond in the same language as the user.
+Generate:
+- Clean
+- Secure
+- Optimized
+- Production-ready code
+Explain briefly what the code does.
+Include best practices.
+Use markdown triple backticks for code blocks.
+"""
+ULTRA_REVIEW_PROMPT = """
+You are a Principal Software Architect and Security Engineer.
+Always respond in the same language as the user.
+Perform an EXTREMELY CRITICAL and PROFESSIONAL code review.
+Analyze:
+1. Logical correctness
+2. Edge cases
+3. Security vulnerabilities (OWASP mindset)
+4. Input validation
+5. Authentication flaws
+6. Concurrency risks
+7. Memory inefficiencies
+8. Performance bottlenecks
+9. Scalability risks
+10. SOLID violations
+11. Clean Architecture violations
+12. Code smells
+13. Dependency risks
+14. Error handling weaknesses
+15. Logging gaps
+16. Deployment risks
+Return structured sections with clear headings.
+"""
+ARCHITECTURE_ANALYSIS_PROMPT = """
+You are a Senior Software Architect.
+Always respond in the same language as the user.
+Analyze the provided system from an ARCHITECTURAL perspective.
+Evaluate:
+- Layered architecture
+- Coupling and cohesion
+- API design
+- Data flow
+- Scalability model
+- Fault tolerance
+- Cloud readiness
+- Observability
+- Caching strategy
+Return structured response with headings.
+"""
+# ===============================
+# FRONTEND (CHAT UI)
+# ===============================
 @app.route("/")
 def index():
     return Response("""
 <html>
 <head>
 <meta charset="UTF-8">
+<title>AI Engineer Assistant</title>
 <style>
+body { margin:0; font-family:Arial; background:#343541; display:flex; justify-content:center; align-items:center; height:100vh; }
+.container { width:900px; height:92vh; 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, select { padding:15px; border:none; outline:none; font-size:16px; }
+input { flex:1; }
+button { padding:15px; border:none; background:#19c37d; font-weight:bold; cursor:pointer; }
+button:hover { opacity:0.9; }
+select { background:#555869; color:white; }
 </style>
 </head>
 <body>
 <div class="container">
     <div id="chat" class="chat"></div>
     <div class="input-area">
+        <select id="mode">
+            <option value="generate">Generate Code</option>
+            <option value="review">Ultra Review</option>
+            <option value="architecture">Architecture Analysis</option>
+        </select>
         <input id="input" placeholder="Write your message..." />
         <button onclick="send()">Send</button>
     </div>
 async function send(){
     const input = document.getElementById("input");
     const chat = document.getElementById("chat");
+    const mode = document.getElementById("mode").value;
     const text = input.value.trim();
     if(!text) return;
         const res = await fetch("/chat",{
             method:"POST",
             headers:{"Content-Type":"application/json"},
+            body:JSON.stringify({message:text, mode:mode})
         });
         const data = await res.json();
 # ===============================
 # API CHAT
 # ===============================
 @app.route("/chat", methods=["POST"])
 def chat():
     try:
         data = request.get_json()
         user_input = data.get("message", "")
+        mode = data.get("mode", "generate")
         if not user_input:
             return jsonify({"error": "Empty message"}), 400
+        # Selección de prompt según modo
+        if mode == "review":
+            system_prompt = ULTRA_REVIEW_PROMPT
+        elif mode == "architecture":
+            system_prompt = ARCHITECTURE_ANALYSIS_PROMPT
+        else:
+            system_prompt = GENERATOR_PROMPT
         prompt = f"{system_prompt}\nUser: {user_input}\nAssistant:"
                 **inputs,
                 max_new_tokens=MAX_NEW_TOKENS,
                 do_sample=True,
+                temperature=0.5,
+                top_p=0.9,
+                repetition_penalty=1.15,
+                no_repeat_ngram_size=3,
                 pad_token_id=tokenizer.eos_token_id
             )
         return jsonify({"error": str(e)}), 500
+# ===============================
+# START SERVER
+# ===============================
 if __name__ == "__main__":
     port = int(os.environ.get("PORT", 7860))
     app.run(host="0.0.0.0", port=port)