Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from extract import extract_domain_data
+from extract import extract_data
 import pandas as pd
 
 def main():
@@ -18,23 +18,28 @@ def main():
     # Aplicar el id al título para que se aplique el estilo CSS
     st.title("PyCARO", anchor="pycaro")
     
-    # Obtener la URL del dominio del input del usuario
-    domain_url = st.text_input("Introduce la URL del dominio ▶", "")
+    # Selector de tipo de búsqueda
+    search_type = st.selectbox("Selecciona el tipo de búsqueda:", ["Dominio", "URL"])
+    
+    # Obtener la URL del dominio o la URL del input del usuario
+    input_label = "Introduce la URL del dominio ▶" if search_type == "Dominio" else "Introduce la URL ▶"
+    user_input = st.text_input(input_label, "")
+    
     if st.button("🔎"):
-        if not domain_url:
+        if not user_input:
             st.warning("No has introducido ninguna URL")
         else:
-            visualize(domain_url)
-  
+            mode = "domain" if search_type == "Dominio" else "url"
+            visualize(user_input, mode)
 
-def visualize(domain_url):  
+def visualize(user_input, mode):  
     try:
         # Fetch and display the website content
         with st.spinner("Loading website data..."):
-            data = extract_domain_data(domain_url)
+            data = extract_data(user_input, mode)
             
             if data:
-                st.subheader(f"Resultados para {domain_url}:")
+                st.subheader(f"Resultados para {user_input}:")
                 df = pd.DataFrame(data)
                 st.table(df)
             else:
@@ -45,4 +50,4 @@ def visualize(domain_url):
         st.error(f"Error: {e}")
 
 if __name__ == "__main__":
-    main()
+    main()