Update app.py

app.py

@@ -8,18 +8,14 @@ import json_repair
 
 # --- 1. CONFIGURATION ---
 print(">>> INITIALIZING SomAI TEXT NODE...")
-
 # Text Model Only
 PHI3_REPO = "microsoft/Phi-3-mini-4k-instruct-gguf"
 PHI3_FILE = "Phi-3-mini-4k-instruct-q4.gguf"
 
 app = FastAPI()
-
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"], 
-    allow_credentials=True,
-    allow_methods=["*"],
+    allow_origins=["*"],
     allow_headers=["*"],
 )
 
@@ -42,7 +38,7 @@ def validate_codes(pipeline):
     for item in pipeline:
         code = item.get('code', '').upper().replace('.', '')
         match = next((k for k in VALID_ICD10 if k.replace('.', '') == code), None)
-        
+
         if match:
             validated.append({
                 "type": item.get("type", "Unknown"),
@@ -89,7 +85,6 @@ class ChatRequest(BaseModel):
     prompt: str
 
 # --- 5. ROUTES ---
-
 @app.get("/")
 def health():
     return {"status": "Active", "system": "SomAI Text Node"}
@@ -97,11 +92,11 @@ def health():
 @app.post("/analyze")
 def analyze(req: AnalysisRequest):
     if not llm: raise HTTPException(503, "Text Model Unavailable")
-    
+
     formatted_prompt = f"<|user|>{req.prompt}<|end|><|assistant|>"
     output = llm(formatted_prompt, max_tokens=400, temperature=0.1, stop=["<|end|>"])
     raw_text = output['choices'][0]['text']
-    
+
     try:
         data = json_repair.repair_json(raw_text, return_objects=True)
     except:
@@ -115,7 +110,7 @@ def analyze(req: AnalysisRequest):
 
     p_code = data.get('primaryConditionCode', {})
     validated_p = validate_codes([{"code": p_code.get('code'), "type": "Primary", "description": p_code.get('description')}])
-    
+
     h_codes = data.get('historyCodes', [])
     validated_h = validate_codes([{"code": h.get('code'), "type": "History", "description": h.get('description')} for h in h_codes])
 
@@ -126,12 +121,12 @@ def analyze(req: AnalysisRequest):
         "historyCodes": validated_h,
         "insuranceNote": data.get("insuranceNote") + " [Validated by SomAI Engine]"
     }
-    
     return final_response
 
 @app.post("/generate")
 def generate(req: ChatRequest):
     if not llm: raise HTTPException(503, "Text Model Unavailable")
+
     formatted_prompt = f"<|user|>{req.prompt}<|end|><|assistant|>"
     output = llm(formatted_prompt, max_tokens=250, temperature=0.6, stop=["<|end|>"])
     return {"text": output['choices'][0]['text'].strip()}