Update app.py

app.py

@@ -3,9 +3,10 @@ from huggingface_hub import InferenceClient
 import plotly.express as px
 from datetime import datetime
 import os
+import json
 
 # ------------------- CONFIG -------------------
-API_KEY = os.environ["HF_API_KEY"]  # Hugging Face secret
+API_KEY = os.environ["HF_API_KEY"]  # Hugging Face secret from Spaces
 MODEL_ID = "duongtruongbinh/smolagents-weather-agent"
 
 client = InferenceClient(API_KEY)
@@ -13,6 +14,7 @@ client = InferenceClient(API_KEY)
 # ------------------- STREAMLIT PAGE -------------------
 st.set_page_config(page_title="Weather Forecast App", page_icon="⛅", layout="wide")
 
+# Dark/Light mode
 mode = st.sidebar.radio("Choose Theme", ["Light", "Dark"])
 if mode == "Dark":
     st.markdown('<style>body {background-color: #0e1117; color: white;}</style>', unsafe_allow_html=True)
@@ -23,42 +25,51 @@ st.markdown("Enter a city name to get current weather and a 5-day forecast")
 city = st.text_input("🌍 Enter city name", "London")
 
 if city:
-    try:
-        response = client(MODEL_ID, {"location": city})  # Correct API call
-    except Exception as e:
-        st.error(f"API Error: {e}")
-    else:
-        if "error" in response:
-            st.error("City not found or API issue. Try again!")
-        else:
-            # Current weather metrics
-            st.subheader(f"Current Weather in {city.title()}")
-            col1, col2, col3, col4 = st.columns(4)
-            with col1:
-                st.metric("🌡 Temperature", f"{response['temperature']} °C")
-            with col2:
-                st.metric("💧 Humidity", f"{response['humidity']} %")
-            with col3:
-                st.metric("🌬 Wind Speed", f"{response['wind_speed']} m/s")
-            with col4:
-                st.image(response['icon_url'])
-
-            st.write(f"**Condition:** {response['condition']}")
-
-            # 5-day forecast chart
-            forecast = response['forecast']
-            dates = [datetime.fromisoformat(f['datetime']).strftime("%d %b") for f in forecast]
-            temps = [f['temp'] for f in forecast]
-
-            fig = px.line(
-                x=dates,
-                y=temps,
-                labels={"x": "Date", "y": "Temperature (°C)"},
-                title=f"5-Day Temperature Forecast for {city.title()}"
-            )
-            fig.update_layout(
-                plot_bgcolor='rgba(0,0,0,0)',
-                paper_bgcolor='rgba(0,0,0,0)' if mode=='Dark' else 'white',
-                font_color='white' if mode=='Dark' else 'black'
-            )
-            st.plotly_chart(fig, use_container_width=True)
+    with st.spinner("Fetching weather data..."):
+        try:
+            # Call Hugging Face text2text API
+            response_text = client.text2text(MODEL_ID, inputs=f"Weather in {city}")["generated_text"]
+            
+            # Attempt to parse JSON from model output
+            try:
+                response = json.loads(response_text)
+            except:
+                st.error("Unable to parse weather data. Model output may not be JSON.")
+                st.write("Raw output:", response_text)
+                response = None
+                
+            if response:
+                # Current weather metrics
+                st.subheader(f"Current Weather in {city.title()}")
+                col1, col2, col3, col4 = st.columns(4)
+                with col1:
+                    st.metric("🌡 Temperature", f"{response['temperature']} °C")
+                with col2:
+                    st.metric("💧 Humidity", f"{response['humidity']} %")
+                with col3:
+                    st.metric("🌬 Wind Speed", f"{response['wind_speed']} m/s")
+                with col4:
+                    st.image(response['icon_url'])
+                
+                st.write(f"**Condition:** {response['condition']}")
+                
+                # 5-day forecast chart
+                forecast = response['forecast']
+                dates = [datetime.fromisoformat(f['datetime']).strftime("%d %b") for f in forecast]
+                temps = [f['temp'] for f in forecast]
+                
+                fig = px.line(
+                    x=dates,
+                    y=temps,
+                    labels={"x": "Date", "y": "Temperature (°C)"},
+                    title=f"5-Day Temperature Forecast for {city.title()}"
+                )
+                fig.update_layout(
+                    plot_bgcolor='rgba(0,0,0,0)',
+                    paper_bgcolor='rgba(0,0,0,0)' if mode=='Dark' else 'white',
+                    font_color='white' if mode=='Dark' else 'black'
+                )
+                st.plotly_chart(fig, use_container_width=True)
+                
+        except Exception as e:
+            st.error(f"API Error: {e}")