# Author: RSK World
# Website: https://rskworld.in
# Email: your.email@example.com
# Year: 2026

from weather_api import get_weather

class Chatbot:
    def get_response(self, user_input):
        # This is a simple placeholder. A more advanced implementation
        # would use a proper NLP library to parse user intent.
        if "weather" in user_input.lower():
            location = self._extract_location(user_input)
            if location:
                weather_data = get_weather(location)
                if "error" in weather_data:
                    return f"Sorry, I couldn't retrieve the weather for {location}. {weather_data['error']}"
                else:
                    return self._format_weather_response(weather_data)
            else:
                return "Please specify a location for the weather forecast."
        else:
            return "I can only provide weather information. Please ask me about the weather."

    def _extract_location(self, user_input):
        # Simple location extraction. This should be improved with a more robust method.
        words = user_input.split()
        for i, word in enumerate(words):
            if word.lower() == "in" and i + 1 < len(words):
                return words[i+1]
        return None

    def _format_weather_response(self, data):
        if "main" in data and "weather" in data:
            city = data.get("name", "Unknown location")
            temp = data["main"].get("temp", "N/A")
            description = data["weather"][0].get("description", "No description")
            return f"The weather in {city} is currently {temp}°C with {description}."
        else:
            return "Sorry, I couldn't understand the weather data."