Update app.py

app.py

@@ -19,34 +19,30 @@ IMAGEGEN_API_URL = "https://api-inference.huggingface.co/models/Artples/LAI-Imag
 # Headers for Hugging Face API requests
 headers = {"Authorization": f"Bearer {HUGGINGFACE_API_KEY}"}
 
-# Inject custom CSS for green theme
+# CSS
 st.markdown("""
     <style>
-    body {
-        background-color: #ffffff;
-    }
-    .stApp {
-        color: #2e7d32;
-        font-family: 'Arial', sans-serif;
-    }
-    .stButton>button {
-        background-color: #66bb6a;
-        color: #fff;
-        font-weight: bold;
-    }
-    .stTextInput>div>input {
-        background-color: #e8f5e9;
-        color: #2e7d32;
-    }
-    .stMarkdown h1, .stMarkdown h2, .stMarkdown h3, .stMarkdown p {
-        color: #388e3c;
-    }
-    .stMarkdown h2 {
-        font-weight: bold;
-    }
+    body {background-color: #ffffff;}
+    .stApp {color: #2e7d32; font-family: 'Arial', sans-serif;}
+    .stButton>button {background-color: #66bb6a; color: #fff; font-weight: bold;}
+    .stTextInput>div>input {background-color: #e8f5e9; color: #2e7d32;}
+    .stMarkdown h1, .stMarkdown h2, .stMarkdown h3, .stMarkdown p {color: #388e3c;}
+    .stMarkdown h2 {font-weight: bold;}
     </style>
 """, unsafe_allow_html=True)
 
+# Initialize session state variables
+if "real_data" not in st.session_state:
+    st.session_state.real_data = {}
+if "story" not in st.session_state:
+    st.session_state.story = ""
+if "music_bytes" not in st.session_state:
+    st.session_state.music_bytes = None
+if "image_bytes" not in st.session_state:
+    st.session_state.image_bytes = None
+if "ngos" not in st.session_state:
+    st.session_state.ngos = []
+
 # Function to fetch weather data
 def fetch_real_data(city: str) -> dict:
     weather_url = f'https://api.openweathermap.org/data/2.5/weather?q={city}&appid={OPENWEATHER_API_KEY}&units=metric'
@@ -61,7 +57,7 @@ def fetch_real_data(city: str) -> dict:
         "weather_condition": weather_data['weather'][0].get('main', 'Data not available')
     }
 
-# Function to determine mood based on weather data
+# Function to determine mood
 def determine_mood(data: dict) -> str:
     weather_condition = data["weather_condition"].lower()
     temperature = data["temperature"]
@@ -94,62 +90,6 @@ def generate_story_with_ai(narrative: str, mood: str) -> str:
     )
     return response.choices[0].message['content'].strip()
 
-# Function to generate simulated environmental data
-def generate_simulated_data(city: str) -> dict:
-    prompt = (
-        f"Generate simulated environmental data for {city} in JSON format with fields:\n"
-        f"1. AQI\n2. Deforestation Rate\n3. Water Quality\n4. Biodiversity Impact"
-    )
-    response = openai.ChatCompletion.create(
-        model="gpt-3.5-turbo",
-        messages=[{"role": "user", "content": prompt}],
-        max_tokens=100,
-        temperature=0.8
-    )
-    response_content = response.choices[0].message['content'].strip()
-    try:
-        return eval(response_content)
-    except Exception as e:
-        st.error(f"Error parsing simulated data: {e}")
-        return {}
-
-# Function to generate music from Hugging Face API
-def generate_music(description: str) -> bytes:
-    payload = {"inputs": description}
-    response = requests.post(MUSICGEN_API_URL, headers=headers, json=payload)
-    if response.status_code != 200:
-        st.error(f"Error generating music: {response.status_code} {response.text}")
-        return None
-    return response.content
-
-# Function to generate an image based on the story
-def generate_image(description: str) -> bytes:
-    payload = {"inputs": description}
-    response = requests.post(IMAGEGEN_API_URL, headers=headers, json=payload)
-    if response.status_code != 200:
-        st.error(f"Error generating image: {response.status_code} {response.text}")
-        return None
-    return response.content
-
-# Function to create a dynamic music description
-def create_music_description(data):
-    mood = data["mood"]
-    weather_condition = data["real_data"]["weather_condition"].lower()
-    temperature = data["real_data"]["temperature"]
-    
-    description = f"{mood} mood with {weather_condition} weather"
-    
-    if temperature < 10:
-        description += " and a cold ambiance"
-    elif 10 <= temperature <= 20:
-        description += " and a cool feel"
-    elif 20 < temperature <= 30:
-        description += " and a warm, lively environment"
-    else:
-        description += " and a hot, energetic vibe"
-    
-    return description
-
 # Function to fetch NGOs using OpenAI
 def fetch_nearby_ngos_with_openai(city: str, interests: list) -> list:
     prompt = (
@@ -173,82 +113,45 @@ def fetch_nearby_ngos_with_openai(city: str, interests: list) -> list:
 st.title("🌿 Eco-Symphony 🎶")
 st.write("Enter a city to explore real-time environmental data, generate AI-created music, and see an AI-generated image based on the story.")
 
-# Input box for city
 city = st.text_input("Enter City Name:", placeholder="Type the name of a city...")
 
-# Generate Button
-if st.button("Generate Environmental Narrative, Music, and Image"):
-    # Fetch real weather data
-    real_data = fetch_real_data(city)
-    
-    if real_data:
-        # Generate narrative and mood
-        narrative = create_narrative(city, real_data)
-        mood = determine_mood(real_data)
-        
-        # Generate AI story
-        story = generate_story_with_ai(narrative, mood)
-
-        # Generate Music and Image Based on Story and Mood
-        music_description = create_music_description({"mood": mood, "real_data": real_data})
-        
-        st.subheader("🎶 Generated Music")
-        st.write(f"Generating music based on: {music_description}")
-        music_bytes = generate_music(music_description)
-        if music_bytes:
-            audio_data = BytesIO(music_bytes)
-            st.audio(audio_data, format="audio/wav")
-
-        st.subheader("🖼️ Generated Image")
-        st.write("Generating image based on the story...")
-        image_bytes = generate_image(story)
-        if image_bytes:
-            image = Image.open(BytesIO(image_bytes))
-            st.image(image, caption="Generated Image based on Story", use_column_width=True)
-
-        # Display Environmental Narrative and Data
-        st.subheader("📜 Environmental Narrative")
-        st.write(narrative)
-        
-        st.subheader("💭 Mood")
-        st.write(f"**Mood**: {mood}")
-
-        st.subheader("🌈 AI-Generated Story")
-        st.write(story)
-
-        # Generate and Display Simulated Environmental Data
-        simulated_data = generate_simulated_data(city)
-        simulated_inner_data = simulated_data.get("data", {})
-        st.subheader("📊 Real Weather Data")
-        st.write("Temperature (°C):", real_data.get("temperature", "Data not available"))
-        st.write("Humidity (%):", real_data.get("humidity", "Data not available"))
-        st.write("Weather Condition:", real_data.get("weather_condition", "Data not available"))
-
-        st.subheader("🧪 Simulated Environmental Data")
-        st.write("AQI:", simulated_inner_data.get("AQI", "Data not available"))
-        st.write("Deforestation Rate:", simulated_inner_data.get("Deforestation Rate", "Data not available"))
-        st.write("Water Quality:", simulated_inner_data.get("Water Quality", "Data not available"))
-        st.write("Biodiversity Impact:", simulated_inner_data.get("Biodiversity Impact", "Data not available"))
-
-        # Collect User's Environmental Interests
-        st.subheader("🌍 Get Involved!")
-        st.write("Choose your areas of interest for saving the environment:")
-        interests = st.multiselect(
-            "Select Areas of Interest:",
-            ["Afforestation", "Water Conservation", "Biodiversity Protection", "Recycling", "Climate Change Awareness"]
-        )
-
-        if st.button("Find Nearby NGOs"):
-            if interests:
-                ngos = fetch_nearby_ngos_with_openai(city, interests)
-                if ngos:
-                    st.subheader("🌟 NGOs Near You")
-                    for ngo in ngos:
-                        st.write(f"**{ngo['name']}**")
-                        st.write(f"📍 Location: {ngo['location']}")
-                        st.write(f"🌱 Focus Area: {ngo['focus']}")
-                        st.write("---")
-                else:
-                    st.write("No NGOs found nearby in your areas of interest.")
-            else:
-                st.warning("Please select at least one area of interest.")
+if st.button("Generate Environmental Data"):
+    st.session_state.real_data = fetch_real_data(city)
+    if st.session_state.real_data:
+        narrative = create_narrative(city, st.session_state.real_data)
+        mood = determine_mood(st.session_state.real_data)
+        st.session_state.story = generate_story_with_ai(narrative, mood)
+
+# Display environmental data
+if st.session_state.real_data:
+    st.subheader("📊 Real Weather Data")
+    st.write("Temperature (°C):", st.session_state.real_data.get("temperature", "Data not available"))
+    st.write("Humidity (%):", st.session_state.real_data.get("humidity", "Data not available"))
+    st.write("Weather Condition:", st.session_state.real_data.get("weather_condition", "Data not available"))
+
+if st.session_state.story:
+    st.subheader("🌈 AI-Generated Story")
+    st.write(st.session_state.story)
+
+# Get User's Environmental Interests
+st.subheader("🌍 Get Involved!")
+st.write("Choose your areas of interest for saving the environment:")
+interests = st.multiselect(
+    "Select Areas of Interest:",
+    ["Afforestation", "Water Conservation", "Biodiversity Protection", "Recycling", "Climate Change Awareness"]
+)
+
+if st.button("Find Nearby NGOs"):
+    if interests:
+        st.session_state.ngos = fetch_nearby_ngos_with_openai(city, interests)
+    else:
+        st.warning("Please select at least one area of interest.")
+
+# Display NGO information
+if st.session_state.ngos:
+    st.subheader("🌟 NGOs Near You")
+    for ngo in st.session_state.ngos:
+        st.write(f"**{ngo['name']}**")
+        st.write(f"📍 Location: {ngo['location']}")
+        st.write(f"🌱 Focus Area: {ngo['focus']}")
+        st.write("---")