Update app.py

app.py

@@ -3,18 +3,16 @@ 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 from Spaces
+API_KEY = os.environ["HF_API_KEY"]
 MODEL_ID = "duongtruongbinh/smolagents-weather-agent"
-
 client = InferenceClient(API_KEY)
+model = client.model(MODEL_ID)  # Get callable model
 
 # ------------------- 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)
@@ -27,19 +25,10 @@ city = st.text_input("🌍 Enter city name", "London")
 if city:
     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
+            response = model({"location": city})  # Correct call
+            if "error" in response:
+                st.error("City not found or API issue. Try again!")
+            else:
                 st.subheader(f"Current Weather in {city.title()}")
                 col1, col2, col3, col4 = st.columns(4)
                 with col1:
@@ -52,12 +41,11 @@ if city:
                     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,
@@ -70,6 +58,6 @@ if city:
                     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}")