Update app.py

app.py

@@ -1,40 +1,91 @@
-from fastapi import FastAPI, WebSocket, WebSocketDisconnect, Request
+import json
+from pathlib import Path
+from datetime import datetime
+
+from fastapi import FastAPI, Request, Form
+from fastapi.responses import RedirectResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
-from typing import List
 
+# --- App Initialization ---
 app = FastAPI()
+
+# Mount static files (for CSS)
 app.mount("/static", StaticFiles(directory="static"), name="static")
 
+# Setup Jinja2 templates
 templates = Jinja2Templates(directory="templates")
 
-class ConnectionManager:
-    def __init__(self):
-        self.active_connections: List[WebSocket] = []
+# --- Data Storage ---
+# Define the path for our temporary data storage
+TMP_DIR = Path("tmp")
+CHAT_FILE = TMP_DIR / "chat_history.json"
 
-    async def connect(self, websocket: WebSocket):
-        await websocket.accept()
-        self.active_connections.append(websocket)
+# Ensure the tmp directory exists
+TMP_DIR.mkdir(exist_ok=True)
 
-    def disconnect(self, websocket: WebSocket):
-        self.active_connections.remove(websocket)
+# --- Helper Functions ---
+def load_chat_history():
+    """Loads chat history from the JSON file."""
+    if CHAT_FILE.exists():
+        with open(CHAT_FILE, "r") as f:
+            try:
+                return json.load(f)
+            except json.JSONDecodeError:
+                return []
+    return []
 
-    async def broadcast(self, message: str):
-        for connection in self.active_connections:
-            await connection.send_text(message)
+def save_chat_history(data):
+    """Saves chat history to the JSON file."""
+    with open(CHAT_FILE, "w") as f:
+        json.dump(data, f, indent=4)
 
-manager = ConnectionManager()
+def get_bot_response(user_message: str) -> str:
+    """A simple rule-based bot response generator."""
+    lower_message = user_message.lower()
+    if "hello" in lower_message or "hi" in lower_message:
+        return "Hello there! How can I help you today?"
+    elif "how are you" in lower_message:
+        return "I'm just a bunch of code, but I'm doing great! Thanks for asking."
+    elif "help" in lower_message:
+        return "I am a simple global chatbot. Just type a message and I'll try to respond. Everyone sees the same conversation."
+    elif "time" in lower_message:
+        return f"The current server time is {datetime.now().strftime('%H:%M:%S')}."
+    else:
+        return f"You said: '{user_message}'"
 
+# --- API Endpoints ---
 @app.get("/")
-async def get(request: Request):
-    return templates.TemplateResponse('index.html', {'request': request})
-
-@app.websocket('/ws')
-async def websocket_endpoint(websocket: WebSocket):
-    await manager.connect(websocket)
-    try:
-        while True:
-            data = await websocket.receive_text()
-            await manager.broadcast(data)
-    except WebSocketDisconnect:
-        manager.disconnect(websocket)
+def read_root(request: Request):
+    """
+    Renders the main chat page.
+    It loads the chat history and passes it to the template.
+    """
+    chat_history = load_chat_history()
+    return templates.TemplateResponse(
+        "index.html", {"request": request, "chat_history": chat_history}
+    )
+
+@app.post("/chat")
+def post_chat_message(message: str = Form(...)):
+    """
+    Handles new chat messages.
+    It saves the user's message and the bot's response to the history.
+    """
+    # Load current history
+    chat_history = load_chat_history()
+
+    # Add user message
+    user_entry = {"sender": "You", "message": message, "timestamp": datetime.now().isoformat()}
+    chat_history.append(user_entry)
+
+    # Get and add bot response
+    bot_message = get_bot_response(message)
+    bot_entry = {"sender": "Bot", "message": bot_message, "timestamp": datetime.now().isoformat()}
+    chat_history.append(bot_entry)
+    
+    # Save updated history
+    save_chat_history(chat_history)
+    
+    # Redirect back to the main page to see the updated chat
+    return RedirectResponse(url="/", status_code=303)