Create app.py

app.py

@@ -0,0 +1,58 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from typing import List, Literal
+
+app = FastAPI(title="Python AI Stub")
+
+class Cursor(BaseModel):
+    l: int
+    c: int
+
+class Viewport(BaseModel):
+    start: int
+    end: int
+    text: str
+
+class Diagnostic(BaseModel):
+    l: int
+    sev: str
+    msg: str
+
+class Memory(BaseModel):
+    short: List[str] = []
+    sess: List[str] = []
+    proj: List[str] = []
+
+class Inp(BaseModel):
+    intent: str
+    file: str
+    lang: str
+    cursor: Cursor
+    viewport: Viewport
+    diag: List[Diagnostic] = []
+    term: str = ""
+    mem: Memory = Memory()
+
+class Need(BaseModel):
+    function: bool = False
+    xrefs: List[str] = []
+    page_ids: List[str] = []
+
+class Out(BaseModel):
+    mode: Literal["patch","full","ask"]
+    patch: str = ""
+    full_text: str = ""
+    explanation: str = ""
+    confidence: float = 0.9
+    need: Need = Need()
+
+@app.post("/code_help", response_model=Out)
+def code_help(x: Inp):
+    before = x.viewport.text
+    if "print(reslt)" in before:
+        after = before.replace("print(reslt)","print(result)",1)
+        h_start = x.viewport.start
+        h_count = x.viewport.end - x.viewport.start + 1
+        patch = f"--- a/{x.file}\n+++ b/{x.file}\n@@ -{h_start},{h_count} +{h_start},{h_count} @@\n-{before}\n+{after}\n"
+        return Out(mode="patch", patch=patch, full_text="", explanation="Fixed misspelling: reslt -> result.", confidence=0.96)
+    return Out(mode="ask", explanation="Need more context.", confidence=0.6)