Update app.py

app.py

@@ -2,58 +2,33 @@ import os
 import requests
 from fastapi import FastAPI, Request
 from fastapi.responses import JSONResponse, HTMLResponse
-from fastapi.middleware.cors import CORSMiddleware
 
 app = FastAPI()
 
-# This allows your index.html to talk to your Python backend
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-# Pulls your secret token from the 'Settings' vault
 HF_TOKEN = os.environ.get("HF_TOKEN")
-
-# TARGET: Your actual Model (without the 's')
-# Change the URL to this:
+# The standard address for a text-generation model
 MODEL_URL = "https://api-inference.huggingface.co/models/danosethrus/EthioDoc"
 
 @app.get("/", response_class=HTMLResponse)
 async def home():
-    try:
-        with open("index.html") as f:
-            return f.read()
-    except FileNotFoundError:
-        return "Error: index.html not found in Files tab!"
+    with open("index.html") as f:
+        return f.read()
 
 @app.post("/ask")
 async def ask_ai(request: Request):
-    user_data = await request.json()
+    data = await request.json()
+    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
     
-    headers = {
-        "Authorization": f"Bearer {HF_TOKEN}",
-        "Content-Type": "application/json"
-    }
-
-    # 'wait_for_model' is the key: it prevents 503 errors while the brain loads
+    # We add 'wait_for_model' so it doesn't error out while loading the 4GB file
     payload = {
-        "inputs": user_data.get("inputs", ""),
+        "inputs": data.get("inputs", ""),
         "options": {"wait_for_model": True}
     }
 
-    try:
-        response = requests.post(MODEL_URL, headers=headers, json=payload, timeout=60)
-        
-        # This helps you see the result in the 'Logs' tab
-        print(f"DEBUG: Status {response.status_code}")
-        print(f"DEBUG: Response {response.text}")
-
-        # If everything is okay, send the AI answer back to the website
-        return JSONResponse(content=response.json(), status_code=response.status_code)
+    response = requests.post(MODEL_URL, headers=headers, json=payload)
+    
+    # This will show the real answer or error in your Logs
+    print(f"DEBUG Status: {response.status_code}")
+    print(f"DEBUG Response: {response.text}")
 
-    except Exception as e:
-        print(f"CRITICAL ERROR: {str(e)}")
-        return JSONResponse({"error": "The bridge is broken. Check Logs."}, status_code=500)
+    return JSONResponse(content=response.json())