Update app.py

app.py

@@ -49,27 +49,28 @@ def search_full_plant_information(plant_name:str):
     """ """
     query = f"{plant_name} plant medicinal uses scientific information site:.edu OR site:.gov OR site:.org"
     search_results=""
-    time.sleep(1)
-    try:
-        search_results = DDGS().text(keywords=query, max_results=5)
-    except Exception as e :
-        pass
-    if search_results:
-        content=""
-        for result in search_results:
-            #content+=requests.get(f"{specialtoken}/Analyze this:{result['body']}").text
-            content+=result['body']+" "
-            content+=fetch_page_content(result['href'])+" "
-            time.sleep(2)
-        prompt = PROMPT_TEMPLATE.format(plant_name=plant_name, content=content)
-        response = requests.get(f"{specialtoken}/{prompt}")
-        #print (response.text)
-        return response.text
-    else:
-        content = DDGS().text(keywords=query, max_results=2)[0]['body']
-        prompt = PROMPT_TEMPLATE.format(plant_name=plant_name, content=content)
-        response = requests.get(f"{specialtoken}/{prompt}")
-        return response.text
+    for attempt in range(3):  # Retry up to 3 times
+        try:
+            search_results = DDGS().text(keywords=query, max_results=5)
+        except DuckDuckGoSearchException as e:
+            if "Ratelimit" in str(e):
+                wait_time = 2 ** attempt  # Exponential backoff
+                print(f"Rate limit hit, retrying in {wait_time} seconds...")
+                time.sleep(wait_time)
+            else:
+                raise
+        raise Exception("Failed to fetch data after retries due to rate limiting.")
+        if search_results:
+            content=""
+            for result in search_results:
+                #content+=requests.get(f"{specialtoken}/Analyze this:{result['body']}").text
+                content+=result['body']+" "
+                content+=fetch_page_content(result['href'])+" "
+                time.sleep(2)
+            prompt = PROMPT_TEMPLATE.format(plant_name=plant_name, content=content)
+            response = requests.get(f"{specialtoken}/{prompt}")
+            #print (response.text)
+            return response.text
     return f"No data Found for {plant_name}!"
 
 DB_NAME="plants.db"