Update app.py

app.py

@@ -5,461 +5,326 @@ import json
 import hashlib
 import uuid
 from datetime import datetime
-import numpy as np
 
-# ============================================
-# STORAGE
-# ============================================
+# ==============================
+# FILE STORAGE
+# ==============================
 
 USERS_FILE = "users.json"
 POSTS_FILE = "posts.json"
 
+os.makedirs("posts", exist_ok=True)
+
 def load_json(file):
     if os.path.exists(file):
-        with open(file, 'r') as f:
-            return json.load(f)
+        try:
+            with open(file) as f:
+                return json.load(f)
+        except:
+            return {}
     return {}
 
-def save_json(file, data):
-    with open(file, 'w') as f:
-        json.dump(data, f, indent=2)
+def save_json(file,data):
+    with open(file,"w") as f:
+        json.dump(data,f,indent=2)
 
 users = load_json(USERS_FILE)
 posts = load_json(POSTS_FILE)
-os.makedirs("posts", exist_ok=True)
 
-# ============================================
-# AUTH FUNCTIONS
-# ============================================
+# ==============================
+# AUTH
+# ==============================
 
-def hash_password(password):
-    return hashlib.sha256(password.encode()).hexdigest()
+def hash_password(p):
+    return hashlib.sha256(p.encode()).hexdigest()
+
+def signup(username,password,display):
 
-def signup(username, password, display_name):
-    if not username or not password:
-        return "❌ Username and password required"
-    
     if username in users:
-        return "❌ Username already exists"
-    
+        return "Username already exists"
+
     users[username] = {
         "password": hash_password(password),
-        "display_name": display_name if display_name else username,
-        "bio": "✨ New to Vibes! Share your first post ✨",
-        "followers": [],
-        "following": [],
-        "created_at": datetime.now().isoformat()
+        "display_name": display or username,
+        "bio":"New to Vibes ✨",
+        "followers":[],
+        "following":[]
     }
-    save_json(USERS_FILE, users)
-    return "✅ Account created! Please login."
 
-def login(username, password):
+    save_json(USERS_FILE,users)
+    return "Account created! Login now."
+
+def login(username,password):
+
     if username not in users:
-        return "❌ User not found", None
-    
+        return "User not found",None
+
     if users[username]["password"] != hash_password(password):
-        return "❌ Wrong password", None
-    
-    return f"✅ Welcome back, {users[username]['display_name']}!", username
-
-# ============================================
-# POST FUNCTIONS - NO PASSWORD NEEDED
-# ============================================
-
-def create_post(username, image, caption, hashtags):
-    if not username:
-        return "❌ Please login first"
-    
+        return "Wrong password",None
+
+    return "Login successful",username
+
+# ==============================
+# POSTS
+# ==============================
+
+def create_post(user,image,caption):
+
+    if user is None:
+        return "Login required"
+
     if image is None:
-        return "❌ Please select an image to upload"
-    
-    try:
-        post_id = str(uuid.uuid4())
-        
-        # Handle image (supports both filepath and numpy array)
-        if isinstance(image, str):
-            img = Image.open(image)
-        else:
-            # Convert numpy array to PIL Image
-            if image.dtype == np.float32:
-                image = (image * 255).astype(np.uint8)
-            img = Image.fromarray(image)
-        
-        # Convert RGBA to RGB if needed
-        if img.mode in ('RGBA', 'LA'):
-            img = img.convert('RGB')
-        
-        # Resize if too large
-        max_size = 1200
-        if max(img.size) > max_size:
-            ratio = max_size / max(img.size)
-            new_size = (int(img.size[0] * ratio), int(img.size[1] * ratio))
-            img = img.resize(new_size, Image.Resampling.LANCZOS)
-        
-        # Save image
-        img.save(f"posts/{post_id}.jpg", "JPEG", quality=85)
-        
-        # Save post data
-        posts[post_id] = {
-            "id": post_id,
-            "username": username,
-            "display_name": users[username]["display_name"],
-            "caption": caption if caption else "",
-            "hashtags": hashtags if hashtags else "",
-            "image": f"posts/{post_id}.jpg",
-            "likes": [],
-            "comments": [],
-            "timestamp": datetime.now().isoformat()
-        }
-        save_json(POSTS_FILE, posts)
-        
-        return "✅ Post shared successfully!"
-    except Exception as e:
-        return f"❌ Error: {str(e)}"
-
-def get_feed():
-    feed_items = []
-    for pid, post in list(posts.items())[-30:]:
-        feed_items.append({
-            "id": pid,
-            "username": post["username"],
-            "display_name": post.get("display_name", post["username"]),
-            "image": post["image"],
-            "caption": post["caption"],
-            "likes": len(post["likes"]),
-            "timestamp": post["timestamp"]
-        })
-    return feed_items[::-1]
-
-def get_user_posts(username):
-    user_posts = []
-    for pid, post in posts.items():
-        if post["username"] == username:
-            user_posts.append(post["image"])
-    return user_posts[-9:]
-
-def get_user_stats(username):
-    if username not in users:
-        return {"display_name": "", "bio": "", "followers": 0, "following": 0, "posts": 0}
-    
-    user = users[username]
-    user_posts = [p for p in posts.values() if p["username"] == username]
-    
-    return {
-        "display_name": user["display_name"],
-        "bio": user.get("bio", "✨ New to Vibes! ✨"),
-        "followers": len(user["followers"]),
-        "following": len(user["following"]),
-        "posts": len(user_posts)
+        return "Upload image"
+
+    post_id=str(uuid.uuid4())
+
+    img = Image.fromarray(image)
+    img = img.convert("RGB")
+
+    path=f"posts/{post_id}.jpg"
+
+    img.save(path,quality=90)
+
+    posts[post_id]={
+        "id":post_id,
+        "username":user,
+        "display":users[user]["display_name"],
+        "caption":caption,
+        "image":path,
+        "likes":[],
+        "comments":[],
+        "time":datetime.now().isoformat()
     }
 
-def update_bio(username, new_bio):
-    if username in users and new_bio:
-        users[username]["bio"] = new_bio
-        save_json(USERS_FILE, users)
-        return "✅ Bio updated!"
-    return "❌ Failed to update bio"
+    save_json(POSTS_FILE,posts)
+
+    return "Post uploaded"
+
+# ==============================
+# FEED
+# ==============================
 
-def search_users(query):
-    if len(query) < 2:
+def load_feed():
+
+    if len(posts)==0:
         return []
-    results = []
-    for username, data in users.items():
-        if query.lower() in username.lower() or query.lower() in data["display_name"].lower():
-            results.append({
-                "username": username,
-                "display_name": data["display_name"],
-                "followers": len(data["followers"])
-            })
-    return results[:10]
-
-# ============================================
-# ORANGE THEME CSS
-# ============================================
-
-custom_css = """
-@import url('https://fonts.googleapis.com/css2?family=Inter:wght@400;500;600;700&display=swap');
-
-* {
-    font-family: 'Inter', sans-serif !important;
-}
-
-.gradio-container {
-    max-width: 1300px !important;
-    margin: 0 auto !important;
-    background: #1a1a1a !important;
-}
-
-.gr-box, .gr-form, .gr-panel {
-    border-radius: 16px !important;
-    border: none !important;
-    background: #2d2d2d !important;
-}
-
-button, .gr-button {
-    border-radius: 40px !important;
-    font-weight: 600 !important;
-    padding: 10px 28px !important;
-    transition: all 0.2s ease !important;
-    font-size: 14px !important;
-    border: none !important;
-}
-
-button:hover, .gr-button:hover {
-    transform: translateY(-2px) !important;
-    filter: brightness(0.9) !important;
-    cursor: pointer !important;
-}
-
-.gr-button-primary {
-    background: linear-gradient(135deg, #FF6B35 0%, #F7931E 100%) !important;
-    color: white !important;
-}
-
-.gr-button-secondary {
-    background: #3d3d3d !important;
-    color: #FF6B35 !important;
-}
-
-input, textarea, .gr-textbox {
-    border-radius: 40px !important;
-    border: 1px solid #3d3d3d !important;
-    background: #2d2d2d !important;
-    padding: 12px 20px !important;
-    font-size: 14px !important;
-    color: #ffffff !important;
-}
-
-input:focus, textarea:focus {
-    border-color: #FF6B35 !important;
-    outline: none !important;
-    background: #3d3d3d !important;
-}
-
-label, .gr-form-label {
-    color: #ffffff !important;
-}
-
-.gallery {
-    border-radius: 16px !important;
-    background: transparent !important;
-    gap: 16px !important;
-}
-
-.tab-nav {
-    background: transparent !important;
-    border-bottom: 1px solid #3d3d3d !important;
-    margin-bottom: 20px !important;
-}
-
-.tab-nav button {
-    font-size: 16px !important;
-    padding: 12px 24px !important;
-    border-radius: 0 !important;
-    background: transparent !important;
-    color: #888888 !important;
-    border-bottom: 2px solid transparent !important;
-}
-
-.tab-nav button.selected {
-    color: #FF6B35 !important;
-    border-bottom: 2px solid #FF6B35 !important;
-}
-
-.profile-card {
-    background: linear-gradient(135deg, #FF6B35 0%, #F7931E 100%);
-    border-radius: 20px;
-    padding: 24px;
-    color: white;
-    text-align: center;
-}
-
-h1, h2, h3, .gr-markdown {
-    color: #ffffff !important;
-}
-
-.gr-markdown p {
-    color: #cccccc !important;
-}
+
+    ordered=sorted(posts.values(),
+                   key=lambda x:x["time"],
+                   reverse=True)
+
+    gallery=[]
+
+    for p in ordered:
+
+        caption=f"""
+{p['display']} (@{p['username']})
+
+{p['caption']}
+
+❤️ {len(p['likes'])} likes
+"""
+
+        gallery.append((p["image"],caption))
+
+    return gallery
+
+# ==============================
+# LIKE
+# ==============================
+
+def like_latest(user):
+
+    if not posts:
+        return "No posts"
+
+    latest=list(posts.keys())[-1]
+
+    if user not in posts[latest]["likes"]:
+        posts[latest]["likes"].append(user)
+
+    save_json(POSTS_FILE,posts)
+
+    return "Liked ❤️"
+
+# ==============================
+# PROFILE
+# ==============================
+
+def load_profile(user):
+
+    if user is None:
+        return "Not logged",[]
+
+    user_posts=[p["image"] for p in posts.values() if p["username"]==user]
+
+    stats=f"""
+### {users[user]["display_name"]}
+
+@{user}
+
+Posts: {len(user_posts)}
+
+Followers: {len(users[user]["followers"])}
+
+Following: {len(users[user]["following"])}
 """
 
-# ============================================
-# BUILD UI
-# ============================================
-
-with gr.Blocks(theme=gr.themes.Soft(), title="Vibes", css=custom_css) as demo:
-    
-    current_user = gr.State(None)
-    
-    # Header
-    gr.HTML("""
-    <div style="text-align: center; padding: 20px 0 10px 0;">
-        <div style="font-size: 48px; margin-bottom: 5px;">📸</div>
-        <h1 style="font-size: 32px; background: linear-gradient(135deg, #FF6B35 0%, #F7931E 100%); -webkit-background-clip: text; -webkit-text-fill-color: transparent; margin: 0;">Vibes</h1>
-        <p style="color: #888888; margin-top: 5px; font-size: 14px;">Share moments. Connect with creators.</p>
-    </div>
-    """)
-    
-    # ===== LOGIN PANEL =====
-    with gr.Column(visible=True, elem_id="login_panel") as login_panel:
-        with gr.Row():
-            with gr.Column(scale=1):
-                pass
-            with gr.Column(scale=2):
-                with gr.Group():
-                    gr.Markdown("### ✨ Welcome to Vibes")
-                    
-                    with gr.Tabs():
-                        with gr.TabItem("🔐 Login"):
-                            login_user = gr.Textbox(label="Username", placeholder="Enter your username")
-                            login_pass = gr.Textbox(label="Password", type="password", placeholder="Enter your password")
-                            login_btn = gr.Button("Login", variant="primary")
-                            login_msg = gr.Markdown("")
-                        
-                        with gr.TabItem("📝 Sign Up"):
-                            signup_user = gr.Textbox(label="Username", placeholder="Choose a username")
-                            signup_pass = gr.Textbox(label="Password", type="password", placeholder="Choose a password")
-                            signup_display = gr.Textbox(label="Display Name", placeholder="What should we call you?")
-                            signup_btn = gr.Button("Create Account", variant="primary")
-                            signup_msg = gr.Markdown("")
-            with gr.Column(scale=1):
-                pass
-    
-    # ===== MAIN APP =====
-    with gr.Column(visible=False, elem_id="main_app") as main_panel:
-        
-        # Top bar
-        with gr.Row():
-            with gr.Column(scale=3):
-                gr.Markdown("")
-            with gr.Column(scale=1):
-                user_display = gr.Markdown("")
-                logout_btn = gr.Button("🚪 Logout", variant="secondary")
-        
-        with gr.Tabs():
-            
-            # Feed Tab
-            with gr.TabItem("🏠 Feed"):
-                feed_gallery = gr.Gallery(
-                    label="", 
-                    columns=3, 
-                    rows=3, 
-                    height=600, 
-                    object_fit="contain",
-                    preview=True
-                )
-                with gr.Row():
-                    refresh_btn = gr.Button("🔄 Refresh", variant="secondary")
-                    feed_info = gr.Markdown("")
-            
-            # Create Tab - NO PASSWORD REQUIRED
-            with gr.TabItem("✨ Create"):
-                gr.Markdown("### Share Your Moment")
-                post_image = gr.Image(label="Upload Image", type="numpy", height=350)
-                post_caption = gr.Textbox(label="Caption", placeholder="What's on your mind?", lines=2)
-                post_hashtags = gr.Textbox(label="Hashtags", placeholder="#inspo #art #photography")
-                post_btn = gr.Button("✨ Share", variant="primary")
-                post_msg = gr.Markdown("")
-            
-            # Profile Tab
-            with gr.TabItem("👤 Profile"):
-                with gr.Row():
-                    with gr.Column(scale=1):
-                        profile_stats = gr.Markdown("")
-                        update_bio_text = gr.Textbox(label="Update Bio", placeholder="Write something about yourself...", lines=3)
-                        update_bio_btn = gr.Button("Update Bio", variant="secondary")
-                        bio_status = gr.Markdown("")
-                    with gr.Column(scale=2):
-                        gr.Markdown("### My Posts")
-                        my_posts_gallery = gr.Gallery(columns=3, rows=2, height=400, object_fit="contain")
-                        refresh_profile_btn = gr.Button("🔄 Refresh", variant="secondary")
-            
-            # Discover Tab
-            with gr.TabItem("🔍 Discover"):
-                gr.Markdown("### Find New Creators")
-                search_input = gr.Textbox(label="Search", placeholder="Enter username...")
-                search_results = gr.Dataframe(
-                    headers=["Username", "Display Name", "Followers"],
-                    datatype=["str", "str", "number"],
-                    interactive=False,
-                    height=300
-                )
-    
-    # ============================================
-    # EVENT HANDLERS
-    # ============================================
-    
-    def handle_signup(user, pwd, display):
-        msg = signup(user, pwd, display)
-        return msg, "", "", ""
-    
-    def handle_login(user, pwd):
-        msg, logged_user = login(user, pwd)
-        if logged_user:
-            stats = get_user_stats(logged_user)
-            return msg, gr.update(visible=False), gr.update(visible=True), logged_user, f"**{stats['display_name']}**\n@{logged_user}"
-        return msg, gr.update(), gr.update(), None, ""
-    
-    def handle_logout():
-        return gr.update(visible=True), gr.update(visible=False), None, ""
-    
-    def handle_create_post(user, img, caption, hashtags):
-        if not user:
-            return "❌ Please login first"
-        msg = create_post(user, img, caption, hashtags)
-        return msg
-    
-    def load_feed():
-        feed = get_feed()
-        if not feed:
-            return [], "✨ No posts yet. Be the first!"
-        images = [f["image"] for f in feed]
-        captions = [f"**{f['display_name']}**\n{f['caption']}\n❤️ {f['likes']} likes" for f in feed]
-        return images, f"📱 {len(feed)} posts"
-    
-    def load_profile(user):
-        if not user:
-            return "Not logged in", []
-        stats = get_user_stats(user)
-        profile_html = f"""
-        <div class="profile-card">
-            <div style="font-size: 48px;">👤</div>
-            <h2>{stats['display_name']}</h2>
-            <p>@{user}</p>
-            <p>{stats['bio']}</p>
-            <div style="display: flex; justify-content: space-around; margin-top: 20px;">
-                <div><strong>{stats['posts']}</strong><br>Posts</div>
-                <div><strong>{stats['followers']}</strong><br>Followers</div>
-                <div><strong>{stats['following']}</strong><br>Following</div>
-            </div>
-        </div>
-        """
-        return profile_html, get_user_posts(user)
-    
-    def handle_update_bio(user, new_bio):
-        if user:
-            return update_bio(user, new_bio)
-        return "Please login"
-    
-    # Wire events
-    signup_btn.click(handle_signup, [signup_user, signup_pass, signup_display], [signup_msg, signup_user, signup_pass, signup_display])
-    login_btn.click(handle_login, [login_user, login_pass], [login_msg, login_panel, main_panel, current_user, user_display])
-    login_btn.click(lambda: load_feed(), None, [feed_gallery, feed_info])
-    login_btn.click(lambda u: load_profile(u), [current_user], [profile_stats, my_posts_gallery])
-    
-    logout_btn.click(handle_logout, None, [login_panel, main_panel, current_user, user_display])
-    
-    post_btn.click(handle_create_post, [current_user, post_image, post_caption, post_hashtags], [post_msg])
-    post_btn.click(lambda: None, None, [post_image, post_caption, post_hashtags])
-    post_btn.click(lambda: load_feed(), None, [feed_gallery, feed_info])
-    post_btn.click(lambda u: load_profile(u), [current_user], [profile_stats, my_posts_gallery])
-    
-    refresh_btn.click(lambda: load_feed(), None, [feed_gallery, feed_info])
-    refresh_profile_btn.click(lambda u: load_profile(u), [current_user], [profile_stats, my_posts_gallery])
-    
-    update_bio_btn.click(handle_update_bio, [current_user, update_bio_text], [bio_status])
-    update_bio_btn.click(lambda u: load_profile(u), [current_user], [profile_stats, my_posts_gallery])
-    
-    search_input.change(search_users, [search_input], [search_results])
-
-demo.launch(server_name="0.0.0.0", server_port=7860)
+    return stats,user_posts
+
+# ==============================
+# SEARCH
+# ==============================
+
+def search_users(q):
+
+    if len(q)<2:
+        return []
+
+    results=[]
+
+    for u,data in users.items():
+
+        if q.lower() in u.lower():
+
+            results.append([
+                u,
+                data["display_name"],
+                len(data["followers"])
+            ])
+
+    return results
+
+# ==============================
+# UI
+# ==============================
+
+with gr.Blocks(title="Vibes") as app:
+
+    current_user=gr.State()
+
+    gr.Markdown("# 📸 Vibes")
+
+    # LOGIN
+    with gr.Tab("Login"):
+
+        user=gr.Textbox(label="Username")
+        pwd=gr.Textbox(label="Password",type="password")
+
+        login_btn=gr.Button("Login")
+
+        login_msg=gr.Textbox()
+
+    # SIGNUP
+    with gr.Tab("Signup"):
+
+        su_user=gr.Textbox(label="Username")
+        su_pwd=gr.Textbox(label="Password",type="password")
+        su_disp=gr.Textbox(label="Display name")
+
+        su_btn=gr.Button("Create account")
+
+        su_msg=gr.Textbox()
+
+    # FEED
+    with gr.Tab("Feed"):
+
+        gallery=gr.Gallery(
+            columns=3,
+            height=600
+        )
+
+        refresh=gr.Button("Refresh")
+
+        like_btn=gr.Button("Like latest")
+
+        like_msg=gr.Textbox()
+
+    # CREATE
+    with gr.Tab("Create Post"):
+
+        img=gr.Image(type="numpy")
+
+        cap=gr.Textbox(label="Caption")
+
+        post_btn=gr.Button("Share")
+
+        post_msg=gr.Textbox()
+
+    # PROFILE
+    with gr.Tab("Profile"):
+
+        profile_stats=gr.Markdown()
+
+        my_gallery=gr.Gallery(columns=3)
+
+        load_profile_btn=gr.Button("Load profile")
+
+    # DISCOVER
+    with gr.Tab("Discover"):
+
+        search=gr.Textbox(label="Search users")
+
+        results=gr.Dataframe(
+            headers=["username","display name","followers"]
+        )
+
+    # =================
+    # EVENTS
+    # =================
+
+    login_btn.click(
+        login,
+        inputs=[user,pwd],
+        outputs=[login_msg,current_user]
+    ).then(
+        load_feed,
+        None,
+        gallery
+    )
+
+    su_btn.click(
+        signup,
+        [su_user,su_pwd,su_disp],
+        su_msg
+    )
+
+    refresh.click(
+        load_feed,
+        None,
+        gallery
+    )
+
+    post_btn.click(
+        create_post,
+        [current_user,img,cap],
+        post_msg
+    ).then(
+        load_feed,
+        None,
+        gallery
+    )
+
+    like_btn.click(
+        like_latest,
+        current_user,
+        like_msg
+    ).then(
+        load_feed,
+        None,
+        gallery
+    )
+
+    load_profile_btn.click(
+        load_profile,
+        current_user,
+        [profile_stats,my_gallery]
+    )
+
+    search.change(
+        search_users,
+        search,
+        results
+    )
+
+app.launch()