Update main.py

main.py

@@ -29,34 +29,12 @@ chatbot = hugchat.ChatBot(cookies=cookies.get_dict())
 ASSISTANT_ID = "66017fca58d60bd7d5c5c26c"  # Replace if needed
 chatbot.new_conversation(assistant=ASSISTANT_ID, switch_to=True)
 
-# FastAPI setup
-app = FastAPI()
-
-# Enable CORS
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-# Request model
-class Question(BaseModel):
-    question: str
-
-# Token stream function
-async def generate_response_stream(prompt: str):
-    for chunk in chatbot.chat(prompt, stream=True):
-        token = chunk.get("token", "")
-        if token:
-            yield token
-            await asyncio.sleep(0.02)
-
-# Endpoint
-@app.post("/ask")
-async def ask(question: Question):
-    return StreamingResponse(
-        generate_response_stream(question.question),
-        media_type="text/plain"
-    )
+# 🔁 Stream response character-by-character as it generates
+print("Assistant:", end=" ", flush=True)
+for token in chatbot.chat("Hello, how can I help you today?", stream=True):
+    print(token, end="", flush=True)
+
+# Optionally: web search example
+# response = chatbot.chat("How many models stored in huggingface?", web_search=True)
+# print("\n\nWeb Search Result:", response.wait_until_done())
+