Update app.py

app.py

@@ -1,40 +1,47 @@
 import streamlit as st
+import requests
 import socket
 
-# Function to resolve the IP address from a URL
-def get_ip_address_from_url(url):
+# Function to get IP address and country name for a given URL
+def get_ip_and_country(url):
     try:
-        # Remove 'http://', 'https://', and 'www.' if present
-        if url.startswith('http://'):
-            url = url[7:]
-        elif url.startswith('https://'):
-            url = url[8:]
+        # Get the IP address of the website
+        ip_address = socket.gethostbyname(url)
         
-        if url.startswith('www.'):
-            url = url[4:]
+        # Query the IP geolocation service (ipinfo.io)
+        response = requests.get(f"https://ipinfo.io/{ip_address}/json")
+        data = response.json()
+
+        # Extract country information
+        country = data.get("country", "Country not available")
         
-        # Get the IP address of the domain
-        ip_address = socket.gethostbyname(url)
-        return ip_address
+        return ip_address, country
     except socket.gaierror:
-        st.error("Invalid URL or Unable to resolve IP address. Please ensure the domain is correct.")
-        return None
+        return None, "Invalid URL or URL not found"
+    except requests.exceptions.RequestException as e:
+        return None, f"Error: {e}"
 
 # Streamlit app
 def main():
-    st.title("Get IP Address from URL")
+    st.title("IP Address and Country Finder")
+    st.write("Enter the URL of the website below to find the IP address and country.")
 
-    # Input field to enter a URL
-    url = st.text_input("Enter the URL of the website (e.g., google.com)")
+    # URL input by the user
+    url = st.text_input("Enter Website URL (e.g., www.example.com)")
 
-    # Button to get the IP address
-    if st.button("Get IP Address"):
+    # Button to submit and fetch details
+    if st.button("Get IP and Country"):
         if url:
-            ip_address = get_ip_address_from_url(url)
+            # Validate and fetch IP and country
+            ip_address, country = get_ip_and_country(url)
+            
             if ip_address:
-                st.success(f"The IP Address for {url} is: {ip_address}")
+                st.success(f"IP Address: {ip_address}")
+                st.success(f"Country: {country}")
+            else:
+                st.error(country)
         else:
-            st.warning("Please enter a valid URL.")
+            st.error("Please enter a valid URL")
 
 if __name__ == "__main__":
     main()