Update app.py

app.py

@@ -1,18 +1,78 @@
 import gradio as gr
-from transformers import pipeline
+from config import SessionManager, validate_all_keys
+from services.market import get_market_data, get_current_price
+from ai.service import AIService
 
-generator = pipeline("text-generation", model="gpt2")
+session = SessionManager()
+ai = AIService()
 
-def suggest_trade(prompt):
-    result = generator(prompt, max_length=100)[0]["generated_text"]
-    return result
 
-iface = gr.Interface(
-    fn=suggest_trade,
-    inputs=gr.Textbox(label="Describe market or ask a crypto trade question"),
-    outputs=gr.Textbox(label="AI-based trading suggestion"),
-    title="Crypto-Walk",
-    description="Enter your market view or question and get AI-generated trading advice."
-)
+def set_keys(openai_key, kucoin_key, kucoin_secret, kucoin_passphrase, provider):
+    session.set_keys(openai_key, kucoin_key, kucoin_secret, kucoin_passphrase, provider)
+    ok, msg = validate_all_keys(session)
+    return msg
 
-iface.launch()
+
+def clear_keys():
+    session.clear()
+    return "✅ Session cleared"
+
+
+def generate(symbol, timeframe, leverage, amount):
+    if not session.is_ready():
+        return {"error": "🔐 Enter API keys first."}
+    market_data = get_market_data(symbol, timeframe, session)
+    current_price = get_current_price(symbol, session)
+    result = ai.generate(
+        symbol=symbol,
+        leverage=leverage,
+        trade_amount=amount,
+        current_price=current_price,
+        market_data=market_data,
+        provider=session.provider,
+        openai_key=session.openai_key,
+        hf_token=session.hf_token,
+    )
+    return (
+        result.to_prompt_dict()
+        if result
+        else {"error": "Failed to generate suggestion"}
+    )
+
+
+with gr.Blocks(theme=gr.themes.Base(default_color_scheme="dark")) as ui:
+    gr.Markdown("# 💹 AI Crypto Trader")
+
+    with gr.Accordion("🔐 API Keys", open=True):
+        openai_key = gr.Textbox(label="OpenAI Key", type="password")
+        kucoin_key = gr.Textbox(label="KuCoin Key", type="password")
+        kucoin_secret = gr.Textbox(label="KuCoin Secret", type="password")
+        kucoin_pass = gr.Textbox(label="KuCoin Passphrase", type="password")
+        provider = gr.Radio(
+            ["OpenAI", "HuggingFace"], value="OpenAI", label="AI Provider"
+        )
+        connect = gr.Button("Connect")
+        status = gr.Textbox(label="Connection Status")
+        connect.click(
+            set_keys,
+            inputs=[openai_key, kucoin_key, kucoin_secret, kucoin_pass, provider],
+            outputs=status,
+        )
+        gr.Button("❌ Clear Session").click(clear_keys, outputs=status)
+
+    gr.Markdown("---")
+
+    with gr.Row():
+        symbol = gr.Textbox(label="Symbol", value="SOLUSDTM")
+        timeframe = gr.Dropdown(
+            ["15m", "1h", "4h", "1d"], label="Timeframe", value="15m"
+        )
+        leverage = gr.Slider(10, 75, step=1, value=20, label="Leverage")
+        amount = gr.Number(label="Trade Amount ($)", value=100)
+
+    output = gr.JSON(label="AI Suggestion")
+    gr.Button("🚀 Generate Suggestion").click(
+        generate, inputs=[symbol, timeframe, leverage, amount], outputs=output
+    )
+
+ui.launch()