Create suggestion.py

suggestion.py

@@ -0,0 +1,60 @@
+from enum import Enum
+from pydantic import BaseModel, Field
+
+
+class TradeDirection(str, Enum):
+    LONG = "long"
+    SHORT = "short"
+
+
+class RecommendationType(str, Enum):
+    RECOMMENDED = "It is recommended to enter the transaction."
+    NOT_RECOMMENDED = "It is not recommended to enter into a transaction."
+    CAUTIOUS = "Entering the trade with caution"
+
+
+class TakeProfitPoints(BaseModel):
+    first: float
+    second: float
+    third: float
+
+
+class TradeSuggestion(BaseModel):
+    symbol: str
+    direction: TradeDirection
+    entry_price: float
+    recommended_leverage: int
+    take_profit: TakeProfitPoints
+    recommendation: RecommendationType
+    current_price: float
+    trade_amount: float
+
+    @property
+    def potential_profit_percentage(self):
+        if self.direction == TradeDirection.LONG:
+            return (
+                ((self.take_profit.third - self.entry_price) / self.entry_price)
+                * 100
+                * self.recommended_leverage
+            )
+        else:
+            return (
+                ((self.entry_price - self.take_profit.third) / self.entry_price)
+                * 100
+                * self.recommended_leverage
+            )
+
+    def to_prompt_dict(self):
+        return {
+            "symbol": self.symbol,
+            "direction": self.direction.value,
+            "entry_price": f"${self.entry_price:.2f}",
+            "leverage": f"{self.recommended_leverage}x",
+            "take_profit_1": f"${self.take_profit.first:.2f}",
+            "take_profit_2": f"${self.take_profit.second:.2f}",
+            "take_profit_3": f"${self.take_profit.third:.2f}",
+            "recommendation": self.recommendation.value,
+            "current_price": f"${self.current_price:.2f}",
+            "trade_amount": f"${self.trade_amount:.2f}",
+            "potential_profit": f"{self.potential_profit_percentage:.2f}%",
+        }