updates

VedAstro · Feb 17, 2025 · eca11a1 · eca11a1
1 parent 1c73533
commit eca11a1
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/demo_all_astro_data.py b/demo_all_astro_data.py
@@ -16,6 +16,10 @@
 #PART 2 : CALCULATE ALL DATA
 #-----------------------------------
 
+#HOROSCOPE PREDICTIONS
+horoscopePredictions = Calculate.HoroscopePredictions(birth_time, "Empty")
+print(json.dumps(horoscopePredictions, indent=4))
+
 #PLANETS
 allPlanetDataList = Calculate.AllPlanetData(PlanetName.Sun, birth_time)
 print(json.dumps(allPlanetDataList, indent=4))

diff --git a/vedastro/calculate.py b/vedastro/calculate.py
@@ -27,7 +27,10 @@ def _make_request(cls, endpoint, params):
             if "Status" in data and data["Status"] == "Fail":
                 print(data["Payload"])
             if "Payload" in data and data["Payload"]:
-                return list(data["Payload"].values())[0]
+                if isinstance(data["Payload"], list):
+                    return data["Payload"]
+                else:
+                    return list(data["Payload"].values())[0]
             else:
                 raise ValueError("Payload is missing or empty")
         else: