Update app.py

app.py

@@ -1,18 +1,109 @@
+import os
+import logging
 from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import JSONResponse, FileResponse
 from fastapi.staticfiles import StaticFiles
-from fastapi.responses import FileResponse
+from dotenv import load_dotenv
+from openai import AsyncOpenAI
 
+# Import your custom modules
+from chainlit.auth import create_jwt
+import chainlit as cl
+import uvicorn
+# Load environment variables from .env file
+load_dotenv()
+
+# Initialize logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)  # Use __name__ to get the root logger
+
+# Initialize FastAPI app
 app = FastAPI()
 
-# Mount the 'static' directory to serve static files.
-# Assuming your static files are in a directory named 'static'.
-app.mount("/static", StaticFiles(directory="static"), name="static")
+# CORS middleware setup
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Specify domains or use ["*"] for open access
+    allow_credentials=True,
+    allow_methods=["*"],  # Specify methods or use ["*"] for all methods
+    allow_headers=["*"],  # Specify headers or use ["*"] for all headers
+)
+
+client = AsyncOpenAI(api_key=os.environ["OPENAI_API_KEY"])
+
+settings = {
+    "model": "gpt-3.5-turbo",
+    "temperature": 0.7,
+    "max_tokens": 500,
+    "top_p": 1,
+    "frequency_penalty": 0,
+    "presence_penalty": 0,
+}
+
+app.mount("/", StaticFiles(directory="static", html=True), name="static")
+logger.info("Static files are being served from the 'static' directory.")
+
+@app.get("/custom-auth")
+async def custom_auth():
+    # Verify the user's identity with custom logic.
+    token = create_jwt(cl.User(identifier="Test User"))
+    logger.info("Custom auth token generated.")
+    return JSONResponse({"token": token})
+
+@cl.on_chat_start
+async def on_chat_start():
+    cl.user_session.set(
+        "message_history",
+        [{"role": "system", "content": "You are a helpful assistant."}],
+    )
+    await cl.Message(content="Connected to Chainlit!").send()
+    logger.info("Chat started with Chainlit.")
+
+@cl.on_message
+async def on_message(message: cl.Message):
+    message_history = cl.user_session.get("message_history")
+    message_history.append({"role": "user", "content": message.content})
 
-@app.get("/")
-def read_root():
-    # Serve your static HTML file at the root.
+    msg = cl.Message(content="")
+    await msg.send()
+
+    stream = await client.chat.completions.create(
+        messages=message_history, stream=True, **settings
+    )
+
+    async for part in stream:
+        if token := part.choices[0].delta.content or "":
+            await msg.stream_token(token)
+
+    message_history.append({"role": "assistant", "content": msg.content})
+    await msg.update()
+    logger.info("Message processed and response sent.")
+
+@app.get("/{full_path:path}")
+async def catch_all(full_path: str):
+    """
+    Catch-all route to serve index.html for any undefined routes,
+    allowing client-side routing to function properly.
+    """
+    logger.info(f"Serving static file for path: {full_path}")
     return FileResponse('static/index.html')
 
-@app.get("/api")
-def read_api():
-    return {"message": "Hello from the FastAPI API!"}
+# from fastapi import FastAPI
+# from fastapi.staticfiles import StaticFiles
+# from fastapi.responses import FileResponse
+
+# app = FastAPI()
+
+# # Mount the 'static' directory to serve static files.
+# # Assuming your static files are in a directory named 'static'.
+# app.mount("/static", StaticFiles(directory="static"), name="static")
+
+# @app.get("/")
+# def read_root():
+#     # Serve your static HTML file at the root.
+#     return FileResponse('static/index.html')
+
+# @app.get("/api")
+# def read_api():
+#     return {"message": "Hello from the FastAPI API!"}