updates

Author: sengiv

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))

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: