Spaces:

Emalawi19
/

Chart

Sleeping

App Files Files Community

Emalawi19 commited on Mar 15

Commit

ae100b3

verified ·

1 Parent(s): 86cc747

Update app.py

Browse files

Files changed (1) hide show

app.py +44 -95

app.py CHANGED Viewed

@@ -1,82 +1,69 @@
-from fastapi import FastAPI, Request, Form, UploadFile, File
 from fastapi.responses import HTMLResponse, RedirectResponse
-from fastapi.staticfiles import StaticFiles
 import sqlite3
-import os
-from datetime import datetime
 app = FastAPI()
-# create folders
-os.makedirs("uploads", exist_ok=True)
-# database
 conn = sqlite3.connect("chat.db", check_same_thread=False)
 c = conn.cursor()
-c.execute("""CREATE TABLE IF NOT EXISTS users(
-username TEXT PRIMARY KEY,
 password TEXT
-)""")
-c.execute("""CREATE TABLE IF NOT EXISTS messages(
-id INTEGER PRIMARY KEY AUTOINCREMENT,
 username TEXT,
-message TEXT,
-image TEXT,
-time TEXT
-)""")
 conn.commit()
-app.mount("/uploads", StaticFiles(directory="uploads"), name="uploads")
-# homepage
 @app.get("/", response_class=HTMLResponse)
 def home():
     return """
     <h1>Emalawi19 Chat</h1>
-    <a href="/login">Login</a><br>
-    <a href="/register">Register</a>
     """
-# register
 @app.get("/register", response_class=HTMLResponse)
-def register_form():
     return """
     <h2>Register</h2>
-    <form action="/register" method="post">
-    Username:<br>
-    <input name="username"><br>
-    Password:<br>
-    <input type="password" name="password"><br><br>
     <button type="submit">Register</button>
     </form>
     """
 @app.post("/register")
 def register(username: str = Form(...), password: str = Form(...)):
-    try:
-        c.execute("INSERT INTO users VALUES(?,?)", (username,password))
-        conn.commit()
-    except:
-        pass
     return RedirectResponse("/login", status_code=302)
-# login
 @app.get("/login", response_class=HTMLResponse)
-def login_form():
     return """
     <h2>Login</h2>
-    <form action="/login" method="post">
-    Username:<br>
     <input name="username"><br>
-    Password:<br>
-    <input type="password" name="password"><br><br>
     <button type="submit">Login</button>
     </form>
     """
 @app.post("/login")
 def login(username: str = Form(...), password: str = Form(...)):
     user = c.execute(
@@ -87,77 +74,39 @@ def login(username: str = Form(...), password: str = Form(...)):
     if user:
         return RedirectResponse(f"/chat/{username}", status_code=302)
-    return HTMLResponse("Login failed")
-# chat page
 @app.get("/chat/{username}", response_class=HTMLResponse)
 def chat(username: str):
-    messages = c.execute(
-        "SELECT username,message,image,time FROM messages"
-    ).fetchall()
     chat_html = ""
-    for u,m,i,t in messages:
-        if m:
-            chat_html += f"<p><b>{u}</b>: {m} ({t})</p>"
-        if i:
-            chat_html += f"<p><b>{u}</b>:<br><img src='/uploads/{i}' width='200'></p>"
     return f"""
-    <html>
-    <head>
-    <meta http-equiv="refresh" content="3">
-    </head>
-    <body>
     <h2>Emalawi19 Chat - {username}</h2>
-    <div style="height:300px;overflow:auto;border:1px solid gray;">
-    {chat_html}
-    </div>
-    <br>
-    <form action="/send/{username}" method="post" enctype="multipart/form-data">
-    <input name="message" placeholder="Type message">
-    <input type="file" name="image">
-    <button type="submit">Send</button>
     </form>
-    </body>
-    </html>
     """
-# send message
-@app.post("/send/{username}")
-async def send(username: str,
-               message: str = Form(""),
-               image: UploadFile = File(None)):
-    filename = None
-    if image:
-        filename = image.filename
-        path = f"uploads/{filename}"
-        with open(path,"wb") as f:
-            f.write(await image.read())
-    time = datetime.now().strftime("%H:%M")
-    c.execute(
-        "INSERT INTO messages(username,message,image,time) VALUES(?,?,?,?)",
-        (username,message,filename,time)
-    )
     conn.commit()
-    return RedirectResponse(f"/chat/{username}", status_code=302)

+from fastapi import FastAPI, Request, Form
 from fastapi.responses import HTMLResponse, RedirectResponse
 import sqlite3
 app = FastAPI()
 conn = sqlite3.connect("chat.db", check_same_thread=False)
 c = conn.cursor()
+c.execute("""
+CREATE TABLE IF NOT EXISTS users(
+username TEXT,
 password TEXT
+)
+""")
+c.execute("""
+CREATE TABLE IF NOT EXISTS messages(
 username TEXT,
+message TEXT
+)
+""")
 conn.commit()
 @app.get("/", response_class=HTMLResponse)
 def home():
     return """
     <h1>Emalawi19 Chat</h1>
+    <a href='/login'>Login</a><br>
+    <a href='/register'>Register</a>
     """
 @app.get("/register", response_class=HTMLResponse)
+def register_page():
     return """
     <h2>Register</h2>
+    <form method="post">
+    <input name="username" placeholder="username"><br>
+    <input name="password" type="password" placeholder="password"><br>
     <button type="submit">Register</button>
     </form>
     """
 @app.post("/register")
 def register(username: str = Form(...), password: str = Form(...)):
+    c.execute("INSERT INTO users VALUES(?,?)",(username,password))
+    conn.commit()
     return RedirectResponse("/login", status_code=302)
 @app.get("/login", response_class=HTMLResponse)
+def login_page():
     return """
     <h2>Login</h2>
+    <form method="post">
     <input name="username"><br>
+    <input name="password" type="password"><br>
     <button type="submit">Login</button>
     </form>
     """
 @app.post("/login")
 def login(username: str = Form(...), password: str = Form(...)):
     user = c.execute(
     if user:
         return RedirectResponse(f"/chat/{username}", status_code=302)
+    return "Login failed"
 @app.get("/chat/{username}", response_class=HTMLResponse)
 def chat(username: str):
+    msgs = c.execute("SELECT username,message FROM messages").fetchall()
     chat_html = ""
+    for u,m in msgs:
+        chat_html += f"<p><b>{u}</b>: {m}</p>"
     return f"""
     <h2>Emalawi19 Chat - {username}</h2>
+    <div>{chat_html}</div>
+    <form action="/send/{username}" method="post">
+    <input name="message">
+    <button>Send</button>
     </form>
     """
+@app.post("/send/{username}")
+def send(username: str, message: str = Form(...)):
+    c.execute("INSERT INTO messages VALUES(?,?)",(username,message))
     conn.commit()
+    return RedirectResponse(f"/chat/{username}", status_code=302)
+# IMPORTANT: start server
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)