Update app.py

app.py

@@ -1,177 +1,108 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 """
-CodeGPS Pro — DeepSeek AI 补丁修复 & 聊天增强版
+CodeGPS Pro — DeepSeek AI 测试安全版
+不在磁盘保存任何凭证；关闭或点击“清除凭证”即忘记。
 """
-import gradio as gr, re, requests, os, json, difflib
-from dataclasses import dataclass
-from typing import List, Dict
+import gradio as gr, json, re, difflib
 
-# ============================================================
-# -------------------- 编辑引擎 -------------------------------
-# ============================================================
-@dataclass
-class CodeBlock:
-    start_line:int; end_line:int; type:str; name:str; content:str
-
-class CodeGPSEngine:
-    def __init__(self): self.reset()
-    def reset(self):
-        self.lines=[]; self.language="unknown"; self.blocks=[]; self.history_stack=[]
-    def load(self,code:str)->str:
-        if not code.strip(): self.reset(); return "⚠️ 代码为空"
-        self.lines=code.split("\n"); self.language=self._detect_lang(code)
-        return f"✅ {self.language.upper()} | {len(self.lines)} 行"
-    def _detect_lang(self,code):
-        pats={'python':[r'def\s','import\s',r':\s*$'],
-              'html':[r'<html',r'<div',r'</\w+>'],
-              'javascript':[r'function\s',r'=>',r'const\s+']}
-        for l,ps in pats.items():
-            if sum(bool(re.search(p,code)) for p in ps)>=2:return l
-        return 'plaintext'
-    def get_code(self): return "\n".join(self.lines)
-    def replace_range(self,s,e,new)->Dict:
-        if not(1<=s<=len(self.lines) and 1<=e<=len(self.lines)): 
-            return {'ok':False,'msg':'⚠️ 行号越界'}
-        self.history_stack.append(self.lines.copy())
-        old=self.lines[s-1:e]
-        self.lines[s-1:e]=new.split("\n")
-        return {'ok':True,'msg':f'✅ 修改 {s}-{e} 行','old':old,'new':new.split("\n"),'s':s,'e':e}
+# ------------------------------------------------------------
+# 极简代码引擎
+# ------------------------------------------------------------
+class Engine:
+    def __init__(self): self.lines=[]; self.history=[]
+    def load(self,code):
+        self.lines=code.split('\n');return f"✅ 载入 {len(self.lines)} 行"
+    def replace_range(self,s,e,new):
+        self.history.append(self.lines.copy())
+        old=self.lines[s-1:e]; self.lines[s-1:e]=new.split('\n')
+        return old
     def undo(self):
-        if not self.history_stack:return{'ok':False,'msg':'无历史记录'}
-        self.lines=self.history_stack.pop();return{'ok':True,'msg':'✅ 已撤销'}
-    def generate_prompt(self,task=''):
-        code_lines="\n".join(f"{i+1:4d}|{l}" for i,l in enumerate(self.lines))
-        return f"语言:{self.language}\n```{self.language}\n{code_lines}\n```\n任务:{task}"
-
-# ============================================================
-# -------------------- DeepSeek API 封装 ----------------------
-# ============================================================
-CONF=os.path.expanduser("~/.codegps_conf.json")
-def load_conf():
-    if os.path.exists(CONF): return json.load(open(CONF))
-    return {"api_key":"","model":"deepseek-chat"}
-def save_conf(c): json.dump(c,open(CONF,"w"),ensure_ascii=False,indent=2)
+        if not self.history:return"无历史"
+        self.lines=self.history.pop();return"✅ 撤销"
+    def get_code(self): return '\n'.join(self.lines)
 
-def call_patch_ai(prompt,model,key,start,end):
-    url="https://api.deepseek.com/chat/completions"
-    sys_prompt=("你是代码修复助手。请只输出JSON补丁，示例："
-                '{"type":"replace","start":行号_int,"end":行号_int,"new_code":"替换内容"}，'
-                "不要包含多余解释。")
-    task_prompt=f"{prompt}\n用户指定修复范围:{start}-{end}行"
-    data={"model":model,
-          "messages":[{"role":"system","content":sys_prompt},
-                      {"role":"user","content":task_prompt}],
-          "stream":False}
-    r=requests.post(url,json=data,
-                    headers={"Authorization":f"Bearer {key}",
-                             "Content-Type":"application/json"})
-    try: return r.json()["choices"][0]["message"]["content"]
-    except Exception: return f"API 返回异常:{r.text}"
+eng=Engine()
 
-def chat_ai(message,model,key,context):
-    url="https://api.deepseek.com/chat/completions"
-    msgs=[{"role":"system","content":"你是编码助手，可连续对话。"}]+context+[{"role":"user","content":message}]
-    r=requests.post(url,json={"model":model,"messages":msgs},
-                    headers={"Authorization":f"Bearer {key}","Content-Type":"application/json"})
-    if r.status_code!=200:return f"错误:{r.status_code}:{r.text}"
-    return r.json()["choices"][0]["message"]["content"]
+# ------------------------------------------------------------
+# 虚拟 AI 调用：你这替换为真实 DeepSeek API 请求
+# ------------------------------------------------------------
+def call_ai_patch(api_key, model, code, task):
+    """这里应调用真实接口；返回 JSON patch 字符串"""
+    # mock：演示只改前两行
+    fake=[{"start":1,"end":2,"new_code":"print('AI已修复')"}]
+    return json.dumps(fake,ensure_ascii=False)
 
-# ============================================================
-# -------------------- diff 生成 ------------------------------
-# ============================================================
-def make_diff(old,new,s,e):
-    diff=list(difflib.unified_diff(old,new,fromfile=f"旧({s}-{e})",tofile=f"新({s}-{e})",lineterm=""))
-    html=["<pre style='background:#fff;color:#000;padding:10px;height:400px;overflow:auto;border:1px solid #ccc;border-radius:6px;'>"]
-    for line in diff:
-        color="red" if line.startswith('-') else "green" if line.startswith('+') else "black"
-        html.append(f"<span style='color:{color}'>{line}</span>")
-    html.append("</pre>")
-    return "".join(html)
+# ------------------------------------------------------------
+# diff 生成
+# ------------------------------------------------------------
+def diff_html(old,new,s,e):
+    out=['<pre style="background:#fff;border:1px solid #ccc;'
+         'padding:8px;height:300px;overflow:auto;">']
+    for l in difflib.unified_diff(old,new,fromfile=f"旧({s}-{e})",
+                                  tofile="新",lineterm=""):
+        c='red'if l.startswith('-')else'green'if l.startswith('+')else'#000'
+        out.append(f'<span style="color:{c}">{l}</span>')
+    out.append('</pre>')
+    return ''.join(out)
 
-# ============================================================
-# -------------------- Gradio UI ------------------------------
-# ============================================================
-def create_ui():
-    eng=CodeGPSEngine(); conf=load_conf()
-    messages=[]
-    with gr.Blocks(title="CodeGPS Pro — DeepSeek AI 补丁修复",theme=gr.themes.Soft()) as app:
-        gr.Markdown("## 🎯 CodeGPS Pro — DeepSeek AI 补丁修复 & 聊天版")
+# ------------------------------------------------------------
+# UI
+# ------------------------------------------------------------
+def make_ui():
+    with gr.Blocks(title="CodeGPS Pro — 安全测试版") as app:
+        gr.Markdown("## 👁 CodeGPS Pro — DeepSeek AI 视觉补丁安全版")
 
-        # API 配置
-        with gr.Accordion("⚙️ DeepSeek 配置",open=False):
-            api=gr.Textbox(label="API KEY",value=conf['api_key'],type="password")
-            model=gr.Textbox(label="模型名",value=conf['model'])
-            save_btn=gr.Button("保存配置"); status=gr.Textbox(label="状态",interactive=False)
-            def save_conf_ui(k,m): save_conf({"api_key":k,"model":m}); return "✅ 已保存"
-            save_btn.click(save_conf_ui,[api,model],[status])
+        # 仅内存保存的 Key / Model
+        creds = gr.State({"api_key":"","model":""})
 
         with gr.Row():
-            # 编辑区
-            with gr.Column(scale=2):
-                code=gr.Code(label="📝 编辑器",language="html",lines=25)
-                info=gr.Textbox(label="状态",interactive=False)
-                with gr.Row():
-                    loadb=gr.Button("加载")
-                    undob=gr.Button("撤销")
-                loadb.click(lambda c:eng.load(c),[code],[info])
-                undob.click(lambda :[eng.undo()['msg'],eng.get_code()],[],[info,code])
-                # 指定行输入
-                with gr.Row():
-                    start_line=gr.Number(label="起始行",value=1,precision=0)
-                    end_line=gr.Number(label="结束行",value=1,precision=0)
-                # 运行html
-                run_html=gr.Button("▶️ 运行 HTML",variant="primary")
-                html_out=gr.HTML(value="<div style='color:#999'>HTML 预览区</div>")
-                def run_html_fn(c):
-                    if "<html" not in c.lower(): return "<div style='color:red'>非 HTML 无法预览</div>"
-                    path="/tmp/preview.html"; open(path,"w",encoding="utf-8").write(c)
-                    return f"<iframe src='file://{path}' style='width:100%;height:90vh;border:none;'></iframe>"
-                run_html.click(run_html_fn,[code],[html_out])
+            code=gr.Code(label="编辑器",language="python",lines=25)
+            chat=gr.Chatbot(label="AI 日志")
+
+        with gr.Accordion("⚙️ 临时凭证设置",open=False):
+            api_in=gr.Textbox(label="API Key",type="password")
+            model_in=gr.Textbox(label="模型名",value="deepseek-chat")
+            save_btn=gr.Button("保存临时凭证")
+            clear_btn=gr.Button("🔐 清除凭证")
+            out_info=gr.Textbox(label="状态",interactive=False)
+
+        # 存入内存状态
+        def save_creds(api,model):
+            return {"api_key":api,"model":model},f"✅ 凭证加载于内存（退出即清除）"
+        def clear_creds():
+            return {"api_key":"","model":""},"🔐 已清除内存凭证"
+        save_btn.click(save_creds,[api_in,model_in],[creds,out_info])
+        clear_btn.click(clear_creds,[],[creds,out_info])
 
-            # AI 区域
-            with gr.Column(scale=1):
-                gr.Markdown("### 🤖 DeepSeek AI 补丁系统")
-                task=gr.Textbox(label="任务说明",lines=4,placeholder="示例: 优化第 行循环性能")
-                gen_btn=gr.Button("🩹 生成补丁")
-                diff_card=gr.HTML(visible=False)
-                with gr.Row(visible=False) as confirm_row:
-                    confirm_btn=gr.Button("✅ 应用补丁")
-                    cancel_btn=gr.Button("❌ 取消")
-                def gen_patch(tasktxt,start,end):
-                    cf=load_conf()
-                    resp=call_patch_ai(eng.generate_prompt(tasktxt),cf['model'],cf['api_key'],start,end)
-                    try: patch=json.loads(resp)
-                    except Exception:
-                        return gr.update(value=f"<pre style='color:red'>{resp}</pre>",visible=True),gr.update(visible=False)
-                    s,e=int(patch.get("start",start)),int(patch.get("end",end))
-                    new=patch.get("new_code","")
-                    eng.pending_patch={'s':s,'e':e,'new':new}
-                    html=make_diff(eng.lines[s-1:e],new.split("\n"),s,e)
-                    return gr.update(value=html,visible=True),gr.update(visible=True)
-                gen_btn.click(gen_patch,[task,start_line,end_line],[diff_card,confirm_row])
-                def apply_patch():
-                    p=eng.pending_patch; r=eng.replace_range(p['s'],p['e'],p['new']); eng.pending_patch=None
-                    return gr.update(value=eng.get_code()),gr.update(value="<div style='color:green'>✅ 已应用</div>",visible=False),gr.update(visible=False)
-                confirm_btn.click(apply_patch,[],[code,diff_card,confirm_row])
-                cancel_btn.click(lambda :[gr.update(visible=False),gr.update(visible=False)],[ ],[diff_card,confirm_row])
+        task=gr.Textbox(label="任务描述",placeholder="讲清要修的部分",lines=2)
+        load_btn=gr.Button("🚀 加载并自动修复")
+        diffview=gr.HTML()
 
-                # 聊天区
-                gr.Markdown("### 💬 与 DeepSeek AI 对话")
-                chatbot=gr.Chatbot(label="DeepSeek 对话区")
-                msg_box=gr.Textbox(label="输入",placeholder="和 DeepSeek AI 交谈…")
-                send_btn=gr.Button("发送")
-                def chat_fn(msg,history):
-                    cf=load_conf()
-                    messages.append({"role":"user","content":msg})
-                    ai_reply=chat_ai(msg,cf['model'],cf['api_key'],messages)
-                    messages.append({"role":"assistant","content":ai_reply})
-                    history.append((msg,ai_reply))
-                    return history,""
-                send_btn.click(chat_fn,[msg_box,chatbot],[chatbot,msg_box])
+        def auto_fix(code_text,task_text,cred,chat_hist):
+            status=eng.load(code_text)
+            if not cred["api_key"]:
+                chat_hist.append(("系统","⚠️ 未设置API Key"))
+                return code_text,chat_hist,"未设置Key",""
+            resp=call_ai_patch(cred["api_key"],cred["model"],code_text,task_text)
+            try:
+                patches=json.loads(resp)
+                for p in sorted(patches,key=lambda x:x["start"],reverse=True):
+                    old=eng.replace_range(p["start"],p["end"],p["new_code"])
+                    html=diff_html(old,p["new_code"].split("\n"),p["start"],p["end"])
+                chat_hist.append(("AI",f"🩹 已自动修复 {len(patches)} 处"))
+                return eng.get_code(),chat_hist,f"{status}|AI 修复完成",html
+            except Exception as e:
+                chat_hist.append(("系统",f"⚠️ AI输出错误 {e}\n{resp}"))
+                return code_text,chat_hist,"出错",""
+        load_btn.click(auto_fix,
+                       inputs=[code,task,creds,chat],
+                       outputs=[code,chat,out_info,diffview])
 
+        undo=gr.Button("↩️ 撤销")
+        undo.click(lambda:[eng.undo(),eng.get_code()],[ ],[out_info,code])
     return app
 
 if __name__=="__main__":
-    app=create_ui()
-    app.launch(server_name="0.0.0.0",server_port=7860,share=True)
+    ui=make_ui()
+    ui.launch(server_port=7860)