From 12524942afeadd21eba7c4ab061b2c1915cd18cb Mon Sep 17 00:00:00 2001
From: Marco Vinciguerra <mvincig11@gmail.com>
Date: Tue, 5 Nov 2024 14:49:50 +0100
Subject: [PATCH] add new import

---
 examples/credits_example.py       |  2 +-
 examples/scrape_example.py        | 22 +++++++++-------------
 examples/scrape_schema_example.py |  3 +--
 scrapegraph_py/__init__.py        |  3 +++
 4 files changed, 14 insertions(+), 16 deletions(-)

diff --git a/examples/credits_example.py b/examples/credits_example.py
index e0bfa24..b53819c 100644
--- a/examples/credits_example.py
+++ b/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()
diff --git a/examples/scrape_example.py b/examples/scrape_example.py
index 68299d4..4dcb1ca 100644
--- a/examples/scrape_example.py
+++ b/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)
diff --git a/examples/scrape_schema_example.py b/examples/scrape_schema_example.py
index 23a908f..4e72811 100644
--- a/examples/scrape_schema_example.py
+++ b/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
diff --git a/scrapegraph_py/__init__.py b/scrapegraph_py/__init__.py
index e69de29..f211dfa 100644
--- a/scrapegraph_py/__init__.py
+++ b/scrapegraph_py/__init__.py
@@ -0,0 +1,3 @@
+from .scrape import scrape
+from .credits import credits
+from .feedback import feedback
\ No newline at end of file