Update amazon_script.py

amazon_script.py

@@ -4,6 +4,7 @@ import json
 from urllib.parse import urlparse
 import asyncio
 import edge_tts
+from gradio_client import Client
 from dotenv import load_dotenv
 
 # Load environment variables
@@ -18,27 +19,24 @@ def extract_asin_from_amazon_url(url):
             return path_parts[i + 1]
     return None
 
-def get_product_details(asin):
-    """Fetches product details using the new API."""
-    url = "https://picfy.xyz/api/amz.php"
-    payload = {"input": asin}
-    headers = {"Content-Type": "application/json"}
-
-    response = requests.post(url, headers=headers, json=payload)
-
-    if response.status_code == 200:
-        return response.json()
-    else:
-        print(f"Error: {response.status_code}, {response.text}")
-        return None
+def get_product_details(asin, country="US"):
+    """Fetches product details using the RapidAPI."""
+    url = os.getenv('RAPIDAPI_URL')
+    querystring = {"asin": asin, "country": country}
+    headers = {
+        "x-rapidapi-key": os.getenv('RAPIDAPI_KEY'),
+        "x-rapidapi-host": os.getenv('RAPIDAPI_HOST')
+    }
+    response = requests.get(url, headers=headers, params=querystring)
+    return response.json() 
 
 def save_product_data(product_data, file_path='data.json'):
     """Saves extracted product data to a JSON file."""
     extracted_data = {
-        "title": product_data.get("title", ""),
-        "description": product_data.get("description", ""),
-        "price": product_data.get("price", ""),
-        "images": product_data.get("images", [])
+        "title": product_data["data"]["product_title"],
+        "description": product_data["data"]["product_description"],
+        "price": product_data["data"]["product_price"],
+        "images": product_data["data"]["product_photos"]  
     }
     with open(file_path, 'w') as json_file:
         json.dump(extracted_data, json_file, indent=4)
@@ -60,24 +58,39 @@ def load_data_from_json(file_path='data.json'):
         return "", ""
 
 def generate_ad_script(title, description, language):
-    """Generates an ad script using the new API."""
-    url = "https://picfy.xyz/api/amz.php"
-    prompt = f"""Create a short advertisement script for:\nTitle: {title}\nDescription: {description}\nLanguage: {language}\nProvide ONLY a JSON response in this exact format:\n{{\n\"ad_script\": \"2 to 3 lines ad script here in the requested language\"\n}}"""
-
-    payload = {"input": prompt}
-    headers = {"Content-Type": "application/json"}
-
-    response = requests.post(url, headers=headers, json=payload)
-
-    if response.status_code == 200:
-        try:
-            return response.json()
-        except json.JSONDecodeError as e:
-            print(f"Error decoding API response: {e}")
-            print("Raw response:", response.text)
-            return {"ad_script": "Error generating ad script"}
-    else:
-        print(f"Error: {response.status_code}, {response.text}")
+    """Generates an ad script using the Groq API."""
+    client = Client(os.getenv('GROQ_API_URL'))
+    prompt = f"""Create a short advertisement script for:
+Title: {title}
+Description: {description}
+Language: {language}
+Provide ONLY a JSON response in this exact format:
+{{
+"ad_script": "2 to 3 lines ad script here in the requested language"
+}}"""
+    
+    result = client.predict(
+        message=prompt,
+        request="llama3-70b-8192",
+        param_3=0.5,
+        param_4=8100,
+        param_5=0.5,
+        param_6=0,
+        api_name="/chat"
+    )
+    
+    try:
+        start_idx = result.find('{')
+        end_idx = result.rfind('}') + 1
+        
+        if start_idx != -1 and end_idx != -1:
+            json_str = result[start_idx:end_idx]
+            return json.loads(json_str)
+        else:
+            raise ValueError("No JSON object found in response")
+    except (json.JSONDecodeError, ValueError) as e:
+        print(f"Error processing API response: {e}")
+        print("Raw response:", result)
         return {"ad_script": "Error generating ad script"}
 
 def save_ad_script(ad_script, file_path='ad_script.txt'):
@@ -94,4 +107,4 @@ async def list_voices():
 async def text_to_speech(voice, text, output_file):
     """Converts text to speech using the selected voice."""
     communicate = edge_tts.Communicate(text, voice)
-    await communicate.save(output_file)
+    await communicate.save(output_file)