minting 2.2.0__tar.gz → 2.2.2__tar.gz

{minting-2.2.0 → minting-2.2.2}/Minting/__init__.py

@@ -1,3 +1,3 @@
 from .client import Client
 
-__version__ = "2.2.0"
+__version__ = "2.2.2"

{minting-2.2.0 → minting-2.2.2}/Minting/client.py

@@ -9,6 +9,11 @@ import sys
 import hashlib
 from datetime import datetime
 from pymongo import MongoClient
+from colorama import Fore, init as colorama_init
+
+# # ----------------- Database Layer -----------------
+# Initialize ANSI support for Windows terminals
+colorama_init(autoreset=True)
 
 
 # ----------------- Database Layer -----------------
@@ -228,7 +233,7 @@ class Client:
                 return pd.DataFrame([{"Error": response_json}])
 
             if not isinstance(response_json, dict):
-                return pd.DataFrame([{
+                return pd.DataFrame([{                               
                     "Error": f"Unexpected response type: {type(response_json)}"
                 }])
 
@@ -241,11 +246,7 @@ class Client:
             for ticker in tickers:
                 ticker_data = result.get(ticker) if isinstance(result, dict) else None
                 if ticker_data is None:
-                    rows.append(pd.DataFrame([{
-                        "Ticker": ticker,
-                        "Parameter": None,
-                        "Error": "No data for ticker"
-                    }]))
+                    print(f"⚠️ No data returned for {ticker}")
                     continue
 
                 for param in parameters:
@@ -255,11 +256,7 @@ class Client:
                         param_data = ticker_data.get(param)
 
                     if param_data is None:
-                        rows.append(pd.DataFrame([{
-                            "Ticker": ticker,
-                            "Parameter": param,
-                            "Error": "No prediction data available"
-                        }]))
+                        print(f"⚠️ {param} data missing for {ticker}")
                         continue
 
                     # Handle both: dict with "data" key OR raw string
@@ -268,29 +265,18 @@ class Client:
                     elif isinstance(param_data, str):
                         raw_data = param_data
                     else:
-                        rows.append(pd.DataFrame([{
-                            "Ticker": ticker,
-                            "Parameter": param,
-                            "Error": f"Unsupported data type: {type(param_data)}"
-                        }]))
+                        print(f"⚠️ Unsupported payload type for {ticker}.{param}: {type(param_data)}")
                         continue
 
                     if not raw_data:
-                        rows.append(pd.DataFrame([{
-                            "Ticker": ticker,
-                            "Parameter": param,
-                            "Error": "Empty prediction data"
-                        }]))
+                        print(f"⚠️ Empty prediction data for {ticker}.{param}")
                         continue
 
                     # Parse CSV-like content: first line header, rest data
                     lines = raw_data.strip().split('\n')
                     if len(lines) < 2:
-                        rows.append(pd.DataFrame([{
-                            "Ticker": ticker,
-                            "Parameter": param,
-                            "Error": "Insufficient data"
-                        }]))
+                        
+                        print(f"⚠️ Insufficient rows for {ticker}.{param}")
                         continue
 
                     data_rows = []
@@ -304,11 +290,7 @@ class Client:
                             })
 
                     if not data_rows:
-                        rows.append(pd.DataFrame([{
-                            "Ticker": ticker,
-                            "Parameter": param,
-                            "Error": "No valid data rows"
-                        }]))
+                        print(f"⚠️ No valid rows parsed for {ticker}.{param}")
                         continue
 
                     df = pd.DataFrame(data_rows)
@@ -316,7 +298,8 @@ class Client:
                     rows.append(df[["Ticker", "Date", "Time", "Predicted Price"]])
 
             if rows:
-                return pd.concat(rows, ignore_index=True)
+                combined = pd.concat(rows, ignore_index=True)
+                return combined[["Ticker", "Date", "Time", "Predicted Price"]]
             else:
                 return pd.DataFrame([{"Error": "No data to display"}])
 
@@ -324,6 +307,37 @@ class Client:
             # last-resort safety
             return pd.DataFrame([{"Error": str(e)}])
 
+    def _render_table(self, df: pd.DataFrame) -> str:
+        """Return a formatted string for the console with a colored header."""
+        if df.empty:
+            return "  No prediction data available."
+
+        headers = ["Ticker", "Date", "Time", "Predicted Price"]
+        values = df[headers].astype(str).values.tolist()
+
+        col_widths = [max(len(str(item)) for item in col) for col in zip(*([headers] + values))]
+
+        spacing = "       "
+        def format_row(row):
+            cells = [f"{str(val):<{width}}" for val, width in zip(row, col_widths)]
+            return spacing.join(cells)
+
+        total_width = sum(col_widths) + len(spacing) * (len(headers) - 1)
+        divider = "=" * total_width
+
+        rows_lines = []
+        for idx, row in enumerate(values):
+            row_line = format_row(row)
+            rows_lines.append(row_line)
+            next_ticker = values[idx + 1][0] if idx + 1 < len(values) else None
+            if next_ticker != row[0]:
+                rows_lines.append("-" * total_width)
+
+        lines = [divider, Fore.CYAN + format_row(headers) + Fore.RESET, divider]
+        lines.extend(rows_lines)
+        lines.append(divider)
+        return "\n".join(lines)
+
     
     def get_prediction(self, tickers, time_frame, parameters, candle="1m"):
         # Normalize tickers
@@ -396,7 +410,7 @@ class Client:
             print("\n" + "="*60)
             print(f"✅ Predictions ({time_frame}, candle={candle})")
             print("="*60)
-            print(df.to_string(index=False))
+            print(self._render_table(df))
             print("="*60)
             print(f"💳 Remaining credits: {remaining}")
             
@@ -436,18 +450,4 @@ class Client:
         return credits
 
 
-# -------------- local test --------------
-# if __name__ == "__main__":
-#     client = Client(
-#         access_token="sk_live_2df1613eca06f07cf0208f8ea30e0ce0f5a50c4a90b0f073321bce1c332e9da2"
-#     )
-
-#     result = client.get_prediction(
-#         tickers=["HDFCBANK"],
-#         time_frame="5 minutes",
-#         parameters=["close"],
-#         candle="1m"
-#     )
-
-#     print("\nRAW RESULT DICT:")
-#     print(result)
+# -------------- local test --------------

{minting-2.2.0 → minting-2.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: minting
-Version: 2.2.0
+Version: 2.2.2
 Summary: Mintzy SDK for stock price prediction
 Author: Om Kulthe
 Author-email: mintzy01.ai@gmail.com

{minting-2.2.0 → minting-2.2.2}/minting.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: minting
-Version: 2.2.0
+Version: 2.2.2
 Summary: Mintzy SDK for stock price prediction
 Author: Om Kulthe
 Author-email: mintzy01.ai@gmail.com

{minting-2.2.0 → minting-2.2.2}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="minting",
-    version="2.2.0",
+    version="2.2.2",
     description="Mintzy SDK for stock price prediction",
     author="Om Kulthe",
     author_email="mintzy01.ai@gmail.com",