Update app.py

app.py

@@ -106,25 +106,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}
@@ -174,15 +155,36 @@ st.write("Enter a city to explore real-time environmental data, generate AI-crea
 
 city = st.text_input("Enter City Name:", placeholder="Type the name of a city...")
 
-if st.button("Generate Environmental Data"):
+if st.button("Generate Environmental Data, Music, and Image"):
     st.session_state.real_data = fetch_real_data(city)
     if st.session_state.real_data:
+        # Generate narrative and mood
         narrative = create_narrative(city, st.session_state.real_data)
         mood = determine_mood(st.session_state.real_data)
+        
+        # Generate AI story
         st.session_state.story = generate_story_with_ai(narrative, mood)
 
-# Display environmental data
+        # Generate Music and Image Based on Story and Mood
+        music_description = f"{mood} mood with {st.session_state.real_data['weather_condition'].lower()} weather"
+        st.session_state.music_bytes = generate_music(music_description)
+        st.session_state.image_bytes = generate_image(st.session_state.story)
+
+# Display Music and Image at the Top
+if st.session_state.music_bytes:
+    st.subheader("🎶 Generated Music")
+    st.audio(BytesIO(st.session_state.music_bytes), format="audio/wav")
+
+if st.session_state.image_bytes:
+    st.subheader("🖼️ Generated Image")
+    st.image(Image.open(BytesIO(st.session_state.image_bytes)), caption="Generated Image based on Story", use_column_width=True)
+
+# Display Environmental Narrative and Data
 if st.session_state.real_data:
+    st.subheader("📜 Environmental Narrative")
+    narrative = create_narrative(city, st.session_state.real_data)
+    st.write(narrative)
+    
     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"))