Update app.py

app.py

@@ -1,13 +1,57 @@
-# scraper.py
-import requests
-from bs4 import BeautifulSoup
+import streamlit as st
 import random
 import json
 from pathlib import Path
+import requests
+from bs4 import BeautifulSoup
 import time
-from transformers import pipeline
 from fake_useragent import UserAgent
-import pandas as pd
+import hashlib
+from datetime import datetime
+import pytz
+
+class SimpleAuth:
+    def __init__(self):
+        self.users_file = Path('users.json')
+        self.init_users_db()
+        
+    def init_users_db(self):
+        if not self.users_file.exists():
+            with open(self.users_file, 'w') as f:
+                json.dump({'users': {}}, f)
+    
+    def get_users(self):
+        with open(self.users_file, 'r') as f:
+            return json.load(f)
+    
+    def save_users(self, users_data):
+        with open(self.users_file, 'w') as f:
+            json.dump(users_data, f)
+    
+    def hash_password(self, password):
+        return hashlib.sha256(password.encode()).hexdigest()
+    
+    def signup(self, email, password):
+        users_data = self.get_users()
+        if email in users_data['users']:
+            return False, "Email already exists"
+        
+        users_data['users'][email] = {
+            'password': self.hash_password(password),
+            'favorites': []
+        }
+        self.save_users(users_data)
+        return True, "Signup successful"
+    
+    def login(self, email, password):
+        users_data = self.get_users()
+        if email not in users_data['users']:
+            return False, "Email not found"
+        
+        if users_data['users'][email]['password'] != self.hash_password(password):
+            return False, "Invalid password"
+        
+        return True, "Login successful"
 
 class QuoteScraper:
     def __init__(self):
@@ -49,64 +93,24 @@ class QuoteScraper:
             print(f"Error scraping BrainyQuote: {e}")
         return quotes
 
-class QuoteGenerator:
-    def __init__(self):
-        self.generator = pipeline('text-generation', model='gpt2')
-        
-    def generate_quote(self, category, max_length=100):
-        prompts = {
-            'motivation': f"Here's an inspiring quote about success and motivation:",
-            'wisdom': f"Here's a profound quote about life and wisdom:",
-            'spiritual': f"Here's a deep spiritual teaching:",
-            'philosophy': f"Here's a philosophical insight about existence:",
-        }
-        
-        prompt = prompts.get(category, f"Here's a quote about {category}:")
-        
-        try:
-            generated_text = self.generator(prompt, max_length=max_length, num_return_sequences=1)[0]['generated_text']
-            # Extract the quote part after the prompt
-            quote = generated_text.split(':')[1].strip()
-            return quote
-        except Exception as e:
-            print(f"Error generating quote: {e}")
-            return None
-
-# app.py
-import streamlit as st
-import firebase_admin
-from firebase_admin import credentials, auth, firestore
-import hashlib
-from datetime import datetime, timedelta
-import pytz
-import base64
-from io import BytesIO
-
-# Initialize Firebase (in production, use environment variables)
-if not firebase_admin._apps:
-    cred = credentials.Certificate("path/to/serviceAccountKey.json")
-    firebase_admin.initialize_app(cred)
-
-db = firestore.client()
-
-def init_session_state():
-    if 'user' not in st.session_state:
-        st.session_state.user = None
-    if 'favorites' not in st.session_state:
-        st.session_state.favorites = []
-
 class QuoteManager:
     def __init__(self):
         self.scraper = QuoteScraper()
-        self.generator = QuoteGenerator()
         self.quotes_file = Path('enhanced_quotes.json')
+        self.favorites_file = Path('favorites.json')
         self.categories = [
             'motivation', 'wisdom', 'spiritual', 'philosophy', 'success',
             'love', 'life', 'happiness', 'science', 'education', 'leadership',
             'buddhist', 'christian', 'islamic', 'hindu', 'jewish',
             'native-american', 'african', 'asian', 'european'
         ]
+        self.init_favorites_db()
         
+    def init_favorites_db(self):
+        if not self.favorites_file.exists():
+            with open(self.favorites_file, 'w') as f:
+                json.dump({}, f)
+    
     def update_quotes_database(self):
         quotes_data = {}
         
@@ -115,13 +119,6 @@ class QuoteManager:
             # Scrape from multiple sources
             quotes.extend(self.scraper.scrape_goodreads(category))
             quotes.extend(self.scraper.scrape_brainyquote(category))
-            
-            # Generate AI quotes
-            for _ in range(5):  # Generate 5 AI quotes per category
-                ai_quote = self.generator.generate_quote(category)
-                if ai_quote:
-                    quotes.append(f"{ai_quote} [AI Generated]")
-                    
             quotes_data[category] = list(set(quotes))  # Remove duplicates
             
         with open(self.quotes_file, 'w', encoding='utf-8') as f:
@@ -135,29 +132,37 @@ class QuoteManager:
         
         with open(self.quotes_file, 'r', encoding='utf-8') as f:
             return json.load(f)
-
-class UserAuth:
-    @staticmethod
-    def signup(email, password):
-        try:
-            user = auth.create_user(
-                email=email,
-                password=password
-            )
-            return user
-        except Exception as e:
-            st.error(f"Error signing up: {e}")
-            return None
-
-    @staticmethod
-    def login(email, password):
-        try:
-            user = auth.get_user_by_email(email)
-            # In production, use proper password verification
-            return user
-        except Exception as e:
-            st.error(f"Error logging in: {e}")
-            return None
+    
+    def add_to_favorites(self, email, quote, category):
+        with open(self.favorites_file, 'r') as f:
+            favorites = json.load(f)
+        
+        if email not in favorites:
+            favorites[email] = []
+            
+        favorites[email].append({
+            'quote': quote,
+            'category': category,
+            'timestamp': datetime.now(pytz.UTC).isoformat()
+        })
+        
+        with open(self.favorites_file, 'w') as f:
+            json.dump(favorites, f)
+    
+    def get_favorites(self, email):
+        with open(self.favorites_file, 'r') as f:
+            favorites = json.load(f)
+        return favorites.get(email, [])
+    
+    def remove_favorite(self, email, quote):
+        with open(self.favorites_file, 'r') as f:
+            favorites = json.load(f)
+            
+        if email in favorites:
+            favorites[email] = [f for f in favorites[email] if f['quote'] != quote]
+            
+        with open(self.favorites_file, 'w') as f:
+            json.dump(favorites, f)
 
 def main():
     st.set_page_config(
@@ -166,17 +171,18 @@ def main():
         layout="wide"
     )
 
-    # Initialize session state and managers
-    init_session_state()
+    # Initialize managers
+    auth_manager = SimpleAuth()
     quote_manager = QuoteManager()
-    quotes_data = quote_manager.load_quotes()
+    
+    # Initialize session state
+    if 'user' not in st.session_state:
+        st.session_state.user = None
 
-    # Custom CSS with enhanced styling
+    # Custom CSS (same as before)
     st.markdown("""
         <style>
-        .main {
-            padding: 2rem;
-        }
+        .main { padding: 2rem; }
         .stButton button {
             width: 100%;
             border-radius: 20px;
@@ -217,21 +223,24 @@ def main():
                 email = st.text_input("Email")
                 password = st.text_input("Password", type="password")
                 if st.button("Login"):
-                    user = UserAuth.login(email, password)
-                    if user:
-                        st.session_state.user = user
-                        st.success("Logged in successfully!")
+                    success, message = auth_manager.login(email, password)
+                    if success:
+                        st.session_state.user = email
+                        st.success(message)
                         st.rerun()
-                        
+                    else:
+                        st.error(message)
             else:
                 email = st.text_input("Email")
                 password = st.text_input("Password", type="password")
                 if st.button("Sign Up"):
-                    user = UserAuth.signup(email, password)
-                    if user:
-                        st.success("Account created successfully!")
+                    success, message = auth_manager.signup(email, password)
+                    if success:
+                        st.success(message)
+                    else:
+                        st.error(message)
         else:
-            st.write(f"Welcome, {st.session_state.user.email}")
+            st.write(f"Welcome, {st.session_state.user}")
             if st.button("Logout"):
                 st.session_state.user = None
                 st.rerun()
@@ -241,6 +250,9 @@ def main():
             ["Home", "Quote Generator", "Browse Categories", "Search Quotes", "My Favorites"]
         )
 
+    # Load quotes
+    quotes_data = quote_manager.load_quotes()
+
     if menu == "Home":
         st.title("Welcome to Global QuoteVerse")
         st.markdown("### Discover Wisdom From All Cultures")
@@ -263,37 +275,21 @@ def main():
     elif menu == "Quote Generator":
         st.title("Generate Custom Quotes")
         
-        col1, col2 = st.columns(2)
-        with col1:
-            category = st.selectbox("Select Category", quote_manager.categories)
-        with col2:
-            source = st.selectbox("Select Source", ["All", "Web Scraped", "AI Generated"])
+        category = st.selectbox("Select Category", quote_manager.categories)
         
         if st.button("Generate Quote"):
-            quotes = quotes_data[category]
-            if source == "AI Generated":
-                quotes = [q for q in quotes if "[AI Generated]" in q]
-            elif source == "Web Scraped":
-                quotes = [q for q in quotes if "[AI Generated]" not in q]
-                
-            if quotes:
-                quote = random.choice(quotes)
-                st.markdown(f"""
-                <div class='quote-box'>
-                    <p>{quote}</p>
-                    <div class='category-tag'>{category}</div>
-                </div>
-                """, unsafe_allow_html=True)
-                
-                if st.session_state.user:
-                    if st.button("Add to Favorites"):
-                        db.collection('favorites').add({
-                            'user_id': st.session_state.user.uid,
-                            'quote': quote,
-                            'category': category,
-                            'timestamp': datetime.now(pytz.UTC)
-                        })
-                        st.success("Added to favorites!")
+            quote = random.choice(quotes_data[category])
+            st.markdown(f"""
+            <div class='quote-box'>
+                <p>{quote}</p>
+                <div class='category-tag'>{category}</div>
+            </div>
+            """, unsafe_allow_html=True)
+            
+            if st.session_state.user:
+                if st.button("Add to Favorites"):
+                    quote_manager.add_to_favorites(st.session_state.user, quote, category)
+                    st.success("Added to favorites!")
 
     elif menu == "Search Quotes":
         st.title("Search Quotes")
@@ -325,30 +321,25 @@ def main():
         else:
             st.title("My Favorite Quotes")
             
-            favorites_ref = db.collection('favorites').where(
-                'user_id', '==', st.session_state.user.uid
-            ).order_by('timestamp', direction=firestore.Query.DESCENDING)
-            
-            favorites = favorites_ref.get()
+            favorites = quote_manager.get_favorites(st.session_state.user)
             
-            for fav in favorites:
-                data = fav.to_dict()
+            for idx, fav in enumerate(favorites):
                 st.markdown(f"""
                 <div class='quote-box'>
-                    <p>{data['quote']}</p>
-                    <div class='category-tag'>{data['category']}</div>
+                    <p>{fav['quote']}</p>
+                    <div class='category-tag'>{fav['category']}</div>
                 </div>
                 """, unsafe_allow_html=True)
                 
                 col1, col2 = st.columns(2)
                 with col1:
-                    if st.button(f"Share {fav.id}", key=f"share_{fav.id}"):
-                        # Generate shareable link
-                        share_url = f"https://yourapp.com/quote/{fav.id}"
-                        st.code(share_url)
+                    if st.button(f"Share", key=f"share_{idx}"):
+                        # Generate shareable text
+                        share_text = f""{fav['quote']}" - via Global QuoteVerse"
+                        st.code(share_text)
                 with col2:
-                    if st.button(f"Remove {fav.id}", key=f"remove_{fav.id}"):
-                        db.collection('favorites').document(fav.id).delete()
+                    if st.button(f"Remove", key=f"remove_{idx}"):
+                        quote_manager.remove_favorite(st.session_state.user, fav['quote'])
                         st.success("Removed from favorites!")
                         st.rerun()