Create app.py

app.py

@@ -0,0 +1,123 @@
+import gradio as gr
+import pymongo
+from bson import ObjectId
+from datetime import datetime
+import bcrypt
+import os
+
+# MongoDB setup
+client = pymongo.MongoClient(os.environ.get("MONGO_URI", "mongodb://localhost:27017/"))
+db = client.aitrendpulse
+
+def register_user(username, password):
+    users = db.users
+    existing_user = users.find_one({"username": username})
+    if existing_user:
+        return "Username already exists"
+    hashed = bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt())
+    user_id = users.insert_one({
+        'username': username,
+        'password': hashed,
+        'created_at': datetime.utcnow()
+    }).inserted_id
+    return f"User registered successfully. User ID: {str(user_id)}"
+
+def login_user(username, password):
+    users = db.users
+    user = users.find_one({'username': username})
+    if user and bcrypt.checkpw(password.encode('utf-8'), user['password']):
+        return f"Login successful. User ID: {str(user['_id'])}"
+    else:
+        return "Invalid username or password"
+
+def get_trends():
+    trends = db.trends.find().sort('created_at', -1).limit(10)
+    return "\n".join([f"Title: {trend['title']}, Category: {trend['category']}" for trend in trends])
+
+def add_trend(title, description, category):
+    trends = db.trends
+    trend_id = trends.insert_one({
+        'title': title,
+        'description': description,
+        'category': category,
+        'created_at': datetime.utcnow()
+    }).inserted_id
+    return f"Trend added successfully. Trend ID: {str(trend_id)}"
+
+def get_preferences(user_id):
+    preferences = db.preferences.find_one({'user_id': ObjectId(user_id)})
+    if preferences:
+        return f"Categories: {', '.join(preferences['categories'])}"
+    else:
+        return "Preferences not found"
+
+def update_preferences(user_id, categories):
+    preferences = db.preferences
+    categories_list = [cat.strip() for cat in categories.split(',')]
+    result = preferences.update_one(
+        {'user_id': ObjectId(user_id)},
+        {'$set': {
+            'categories': categories_list,
+            'updated_at': datetime.utcnow()
+        }},
+        upsert=True
+    )
+    if result.modified_count > 0 or result.upserted_id:
+        return "Preferences updated successfully"
+    else:
+        return "No changes made to preferences"
+
+# Gradio interface
+with gr.Blocks() as demo:
+    gr.Markdown("# AITrendPulse")
+    
+    with gr.Tab("User Management"):
+        with gr.Group():
+            gr.Markdown("## Register User")
+            register_username = gr.Textbox(label="Username")
+            register_password = gr.Textbox(label="Password", type="password")
+            register_button = gr.Button("Register")
+            register_output = gr.Textbox(label="Result")
+            register_button.click(register_user, inputs=[register_username, register_password], outputs=register_output)
+        
+        with gr.Group():
+            gr.Markdown("## Login User")
+            login_username = gr.Textbox(label="Username")
+            login_password = gr.Textbox(label="Password", type="password")
+            login_button = gr.Button("Login")
+            login_output = gr.Textbox(label="Result")
+            login_button.click(login_user, inputs=[login_username, login_password], outputs=login_output)
+    
+    with gr.Tab("Trend Management"):
+        with gr.Group():
+            gr.Markdown("## Get Trends")
+            get_trends_button = gr.Button("Get Trends")
+            trends_output = gr.Textbox(label="Trends")
+            get_trends_button.click(get_trends, inputs=[], outputs=trends_output)
+        
+        with gr.Group():
+            gr.Markdown("## Add Trend")
+            trend_title = gr.Textbox(label="Title")
+            trend_description = gr.Textbox(label="Description")
+            trend_category = gr.Textbox(label="Category")
+            add_trend_button = gr.Button("Add Trend")
+            add_trend_output = gr.Textbox(label="Result")
+            add_trend_button.click(add_trend, inputs=[trend_title, trend_description, trend_category], outputs=add_trend_output)
+    
+    with gr.Tab("User Preferences"):
+        with gr.Group():
+            gr.Markdown("## Get Preferences")
+            get_pref_user_id = gr.Textbox(label="User ID")
+            get_pref_button = gr.Button("Get Preferences")
+            get_pref_output = gr.Textbox(label="Preferences")
+            get_pref_button.click(get_preferences, inputs=[get_pref_user_id], outputs=get_pref_output)
+        
+        with gr.Group():
+            gr.Markdown("## Update Preferences")
+            update_pref_user_id = gr.Textbox(label="User ID")
+            update_pref_categories = gr.Textbox(label="Categories (comma-separated)")
+            update_pref_button = gr.Button("Update Preferences")
+            update_pref_output = gr.Textbox(label="Result")
+            update_pref_button.click(update_preferences, inputs=[update_pref_user_id, update_pref_categories], outputs=update_pref_output)
+
+demo.launch()