Update app.py

app.py

@@ -1,111 +1,111 @@
-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 dotenv import load_dotenv
-from openai import AsyncOpenAI
+# 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 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()
+# # 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 logging
+# logging.basicConfig(level=logging.INFO)
+# logger = logging.getLogger(__name__)  # Use __name__ to get the root logger
 
-# Initialize FastAPI app
-app = FastAPI()
+# # Initialize FastAPI app
+# app = FastAPI()
 
-# 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.")
+# # 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("/")
-def read_root():
-    """
-    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")
-    return FileResponse('static/index.html')
+# @app.get("/")
+# def read_root():
+#     """
+#     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")
+#     return FileResponse('static/index.html')
     
-@app.get("/api")
-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})
+# @app.get("/api")
+# 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_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})
+# @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})
 
-    msg = cl.Message(content="")
-    await msg.send()
+#     msg = cl.Message(content="")
+#     await msg.send()
 
-    stream = await client.chat.completions.create(
-        messages=message_history, stream=True, **settings
-    )
+#     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)
+#     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.")
+#     message_history.append({"role": "assistant", "content": msg.content})
+#     await msg.update()
+#     logger.info("Message processed and response sent.")
 
 
 
-# from fastapi import FastAPI
-# from fastapi.staticfiles import StaticFiles
-# from fastapi.responses import FileResponse
+from fastapi import FastAPI
+from fastapi.staticfiles import StaticFiles
+from fastapi.responses import FileResponse
 
-# app = FastAPI()
+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")
+# 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("/")
+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!"}
+@app.get("/api")
+def read_api():
+    return {"message": "Hello from the FastAPI API!"}