Update blaxel_main.py

blaxel_main.py

@@ -1,62 +1,26 @@
-import asyncio
-import json
+# blaxel_main.py (Smoke Test)
 from fastapi import FastAPI
 from pydantic import BaseModel
 from sse_starlette.sse import EventSourceResponse
+import asyncio
 
-# --- API Models ---
-# This defines the data structure we expect from our Gradio app
-class ProblemRequest(BaseModel):
-    problem: str
-    run_count: int
-    do_calibrate: bool
-
-# --- FastAPI App ---
-# This creates our web application
 app = FastAPI()
 
-# --- API Endpoints ---
-
-@app.get("/")
-def read_root():
-    """
-    A simple "health check" endpoint.
-    If we can see this in a browser, we know our Blaxel server is running.
-    """
-    return {"status": "MudabbirAI Backend is running."}
-
+# Input model (matches Gradio's request)
+class ProblemRequest(BaseModel):
+    problem: str
 
 @app.post("/solve_problem")
 async def solve_problem(request: ProblemRequest):
     """
-    This is the main endpoint our Gradio app calls.
-    It streams back a simple "hello world" test to confirm the connection.
+    A simple streaming endpoint to confirm the connection.
     """
     
-    # This is the streaming function
-    async def stream_solution():
-        try:
-            # 1. Acknowledge the request
-            yield f"[1/6] 🧠 Agent activated. Connection to Blaxel backend successful."
-            await asyncio.sleep(1) # simulate work
-            
-            # 2. Show the data we received
-            yield f"Received Problem: {request.problem}"
-            await asyncio.sleep(1)
-            yield f"Received Run Count: {request.run_count}"
-            await asyncio.sleep(1)
-            
-            # 3. Send the final "package" (as a string)
-            # We use a special "FINAL:" prefix
-            final_package = {
-                "text": "## Test Successful!\n\nThe Blaxel backend is live. We are now ready to build the real agent logic.",
-                "audio": None
-            }
-            yield f"FINAL:{json.dumps(final_package)}"
-        
-        except Exception as e:
-            # Send any errors as part of the stream
-            yield f"Backend Error: {str(e)}"
+    async def stream_test():
+        yield f"Connection established for problem: '{request.problem}'"
+        await asyncio.sleep(0.5)
+        yield "MudabbirAI is thinking..."
+        await asyncio.sleep(0.5)
+        yield "Test complete. The connection is good!"
 
-    # Return the streaming response
-    return EventSourceResponse(stream_solution())
+    return EventSourceResponse(stream_test())