Update app.py

app.py

@@ -1,108 +1,52 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 """
-CodeGPS Pro — DeepSeek AI 测试安全版
-不在磁盘保存任何凭证；关闭或点击“清除凭证”即忘记。
+CodeGPS Pro — DeepSeek 两窗交互版
+左：代码编辑器   右：AI 聊天（上下文连续）
 """
-import gradio as gr, json, re, difflib
-
-# ------------------------------------------------------------
-# 极简代码引擎
-# ------------------------------------------------------------
-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:return"无历史"
-        self.lines=self.history.pop();return"✅ 撤销"
-    def get_code(self): return '\n'.join(self.lines)
-
-eng=Engine()
-
-# ------------------------------------------------------------
-# 虚拟 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 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)
-
-# ------------------------------------------------------------
-# UI
-# ------------------------------------------------------------
-def make_ui():
-    with gr.Blocks(title="CodeGPS Pro — 安全测试版") as app:
-        gr.Markdown("## 👁 CodeGPS Pro — DeepSeek AI 视觉补丁安全版")
-
-        # 仅内存保存的 Key / Model
-        creds = gr.State({"api_key":"","model":""})
+import gradio as gr, json
+
+# 模拟本地修复逻辑；接通真实 API 时替换此函数
+def call_deepseek(api_key, model, full_prompt):
+    # 在这里执行真实网络调用，返回生成内容
+    fake_reply = "我已看见你的全部代码，这里只是示例回答。\n" \
+                 "若需要修改，请告诉我行号或描述问题。"
+    return fake_reply
+
+# 聊天处理函数
+def chat_round(message, history, code_text, api_key, model):
+    if not api_key.strip():
+        return history + [("系统", "⚠️ 请先输入 API Key")], ""
+    # 构造 prompt，把最新代码附加进去
+    prompt = f"用户消息：{message}\n\n当前完整代码:\n```{code_text}```"
+    ai_reply = call_deepseek(api_key, model, prompt)
+    history.append((message, ai_reply))
+    return history, ""
+
+# Gradio UI
+def create_ui():
+    with gr.Blocks(title="CodeGPS Pro – DeepSeek 两窗版") as app:
+        gr.Markdown("## 👁 CodeGPS Pro — DeepSeek AI 聊天修复版")
+
+        with gr.Accordion("🔐 临时配置", open=False):
+            api_in = gr.Textbox(label="API KEY", type="password")
+            model_in = gr.Textbox(label="模型名", value="deepseek-chat")
 
         with gr.Row():
-            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])
+            code = gr.Code(label="编辑器", language="python", lines=30)
+            chatbot = gr.Chatbot(label="DeepSeek 对话")
 
-        task=gr.Textbox(label="任务描述",placeholder="讲清要修的部分",lines=2)
-        load_btn=gr.Button("🚀 加载并自动修复")
-        diffview=gr.HTML()
+        msg = gr.Textbox(label="输入问题或指令", placeholder="和 DeepSeek AI 对话…")
+        send = gr.Button("发送")
 
-        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])
+        send.click(
+            chat_round,
+            inputs=[msg, chatbot, code, api_in, model_in],
+            outputs=[chatbot, msg]
+        )
 
-        undo=gr.Button("↩️ 撤销")
-        undo.click(lambda:[eng.undo(),eng.get_code()],[ ],[out_info,code])
     return app
 
-if __name__=="__main__":
-    ui=make_ui()
+if __name__ == "__main__":
+    ui = create_ui()
     ui.launch(server_port=7860)