Update app.py

app.py

@@ -1,24 +1,21 @@
 import streamlit as st
 import random
-from datasets import load_dataset
-from transformers import pipeline
-from dataset import load_user_quotes, save_user_quote, upvote_quote
+import openai
 from PIL import Image
 import requests
 
-# Load predefined quotes dataset
-dataset = load_dataset("Abirate/english_quotes")
-random_quote = dataset['train'][random.randint(0, len(dataset['train'])-1)]
-random_text = random_quote['quote']
-random_author = random_quote['author']
+# Set your OpenAI API key
+openai.api_key = "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"  # Replace with your OpenAI API key
+
+# Load predefined quotes dataset (Here, using a static example for now)
+random_text = "The only way to do great work is to love what you do."
+random_author = "Steve Jobs"
 
 # Load user-submitted quotes
 if "user_quotes" not in st.session_state:
-    st.session_state.user_quotes = load_user_quotes()
-
-# Load AI quote generator (GPT-2)
-quote_generator = pipeline("text-generation", model="gpt2")
+    st.session_state.user_quotes = []
 
+# Set up Streamlit page
 st.set_page_config(page_title="Quotation of the Day", page_icon="📝", layout="centered")
 st.title("📜 Quotation of the Day")
 
@@ -28,26 +25,40 @@ st.write(f"🗣️ *{random_text}* – {random_author}")
 # AI-Generated Quote Image
 st.subheader("🖼️ AI-Generated Quote Image")
 if st.button("Generate Quote Image"):
-    response = requests.get(f"https://api-inference.huggingface.co/models/dalle-mini", 
-                            headers={"Authorization": "Bearer YOUR_HF_API_KEY"}, 
-                            json={"inputs": random_text})
-    if response.status_code == 200:
-        image = Image.open(response.content)
-        st.image(image, caption="AI-Generated Quote Image")
-    else:
-        st.error("Failed to generate image.")
+    # Call OpenAI's DALL·E model to generate an image from the quote
+    response = openai.Image.create(
+        prompt=random_text,  # Use the random quote as the prompt
+        n=1,  # Generate 1 image
+        size="1024x1024"  # Image size
+    )
+    
+    # Extract the image URL from the response
+    image_url = response['data'][0]['url']
+    
+    # Display the image
+    st.image(image_url, caption="AI-Generated Quote Image")
 
-# AI-Generated Quote
+# AI-Generated Quote (Using GPT-3)
 st.subheader("💡 Get an AI-Generated Quote")
 if st.button("Generate AI Quote"):
-    ai_quote = quote_generator("A great quote is", max_length=50, num_return_sequences=1)[0]["generated_text"]
+    # Call OpenAI's GPT-3 model to generate a quote
+    response = openai.Completion.create(
+        engine="text-davinci-003",  # Choose the GPT-3 model
+        prompt="A great quote is",  # Starting text for the quote
+        max_tokens=50  # Limit the number of tokens (words/characters)
+    )
+    
+    # Get the generated quote
+    ai_quote = response.choices[0].text.strip()
+    
+    # Display the AI-generated quote
     st.write(f"✨ *{ai_quote}* – AI")
 
-# User Authentication with Hugging Face
-st.subheader("🔑 Login with Hugging Face")
+# User Authentication with Hugging Face (In this case, we’re not using it)
+st.subheader("🔑 Submit Your Own Quote")
 if "username" not in st.session_state:
-    username = st.text_input("Enter Hugging Face username (for quote submission)")
-    if st.button("Login"):
+    username = st.text_input("Enter your name (for quote submission)")
+    if st.button("Submit"):
         st.session_state.username = username
         st.success(f"✅ Logged in as {username}")
 
@@ -55,25 +66,21 @@ if "username" not in st.session_state:
 if "username" in st.session_state:
     st.subheader("📝 Submit Your Own Quote")
     user_quote = st.text_area("Enter your quote")
-    user_name = st.text_input("Your Name (Leave blank for Anonymous)")
-
-    if st.button("Submit"):
+    
+    if st.button("Submit Quote"):
         if user_quote.strip():
-            save_user_quote(user_quote, user_name if user_name else "Anonymous")
-            st.session_state.user_quotes = load_user_quotes()  # Update session state
+            st.session_state.user_quotes.append({'quote': user_quote, 'author': st.session_state.username})
             st.success("✅ Quote added successfully!")
-            st.rerun()  # Use st.rerun() instead of st.experimental_rerun()
         else:
             st.error("❌ Please enter a quote!")
 
-# Community Quotes with Upvotes
+# Display community quotes (user-submitted quotes)
 st.subheader("📖 Community Quotes")
-for i, q in enumerate(st.session_state.user_quotes):  # Use session state for user_quotes
+for i, q in enumerate(st.session_state.user_quotes):
     col1, col2 = st.columns([4, 1])
     with col1:
-        st.write(f"💬 *{q['quote']}* – {q['author']}  | 👍 {q['upvotes']} upvotes")
+        st.write(f"💬 *{q['quote']}* – {q['author']}")
     with col2:
-        if st.button(f"Upvote {i}", key=f"upvote_{i}"):  # Indented correctly now
-            upvote_quote(i)
-            st.session_state.user_quotes = load_user_quotes()  # Update session state
-            st.rerun()  # Use st.rerun() here as well
+        if st.button(f"Upvote {i}", key=f"upvote_{i}"):
+            st.session_state.user_quotes[i]['upvotes'] = st.session_state.user_quotes.get('upvotes', 0) + 1
+            st.experimental_rerun()