add new import

examples/credits_example.py

@@ -5,7 +5,7 @@
 
 import os
 from dotenv import load_dotenv
-from scrapegraph_py.credits import credits
+from scrapegraph_py import credits
 
 # Load environment variables from a .env file
 load_dotenv()

examples/scrape_example.py

@@ -1,16 +1,12 @@
+import os
 from scrapegraph_py.scrape import scrape
-from dotenv import load_dotenv  # Import load_dotenv
-import os  # Import os to access environment variables
-import json  # Import json for beautifying output
+from dotenv import load_dotenv
 
-def main():
-    """Main function to execute the scraping process."""
-    load_dotenv()
-    api_key = os.getenv("SCRAPEGRAPH_API_KEY")
-    url = "https://scrapegraphai.com/"
-    prompt = "What does the company do?"
 
-    result = scrape(api_key, url, prompt)
-    print(result)
-if __name__ == "__main__":
-    main()
+load_dotenv()
+api_key = os.getenv("SCRAPEGRAPH_API_KEY")
+url = "https://scrapegraphai.com/"
+prompt = "What does the company do?"
+
+result = scrape(api_key, url, prompt)
+print(result)

examples/scrape_schema_example.py

@@ -1,9 +1,8 @@
+import os
 from pydantic import BaseModel, Field
 from scrapegraph_py.scrape import scrape
 from dotenv import load_dotenv
-import os
 
-# Load environment variables from .env file
 load_dotenv()
 
 # Define a Pydantic schema

scrapegraph_py/__init__.py

@@ -0,0 +1,3 @@
+from .scrape import scrape
+from .credits import credits
+from .feedback import feedback