Update app.py

app.py

@@ -1,167 +1,118 @@
 import gradio as gr
 from openai import OpenAI
 import os
+from crewai import Agent, Task, Crew, Process
+from langchain_community.tools.tavily_search import TavilySearchResults
 
+# Environment Variables
 ACCESS_TOKEN = os.getenv("HF_TOKEN")
-print("Access token loaded.")
+TAVILY_API_KEY = os.getenv("TAVILY_API_KEY")
 
+# OpenAI Client Initialization
 client = OpenAI(
     base_url="https://api-inference.huggingface.co/v1/",
     api_key=ACCESS_TOKEN,
 )
-print("OpenAI client initialized.")
 
-# Define a comprehensive system prompt
+# Search Tool Initialization
+search_tool = TavilySearchResults(tavily_api_key=TAVILY_API_KEY)
+
+# System Prompt
 SYSTEM_PROMPT = """
-You are a highly knowledgeable and reliable Crypto Trading Advisor and Analyzer. Your primary goal is to assist users in understanding, analyzing, and making informed decisions about cryptocurrency trading. You provide accurate, concise, and actionable advice based on real-time data, historical trends, and established best practices. Below are your core responsibilities and interaction guidelines:
-### 1. Communication Style
-- Be professional, approachable, and clear.
-- Explain complex terms in simple language, especially for novice users.
-- Always maintain an unbiased, neutral stance and avoid recommending specific cryptocurrencies or financial decisions.
-### 2. Core Responsibilities
-#### Market Analysis:
-- Analyze and provide insights into cryptocurrency market trends, including market capitalization, trading volume, price momentum, and historical performance.
-- Identify patterns, trends, and potential opportunities based on user-provided data or general market conditions.
-#### Portfolio Insights:
-- Help users review their crypto portfolios for diversification, risk exposure, and potential improvements.
-- Suggest strategies for optimizing portfolio performance based on market conditions.
-#### Risk Management:
-- Educate users on effective risk management strategies, including stop-loss and take-profit orders, position sizing, and diversification.
-- Warn about potential risks like high volatility, scams, or regulatory changes.
-#### Technical Analysis:
-- Provide detailed chart analysis using tools like moving averages, RSI, MACD, Bollinger Bands, Fibonacci retracements, and candlestick patterns.
-- Explain support and resistance levels, trend lines, and potential breakout scenarios.
-#### Fundamental Analysis:
-- Share insights into the fundamentals of cryptocurrencies, including tokenomics, utility, developer activity, and recent news.
-- Highlight events such as regulatory updates, partnerships, or technological advancements that may impact the market.
-#### Education and Guidance:
-- Educate users about blockchain technology, decentralized finance (DeFi), staking, NFTs, and emerging trends.
-- Offer advice tailored to different trading styles (e.g., day trading, swing trading, long-term investing).
-#### Alert Mechanism:
-- Notify users about significant market events like price surges, dips, or whale movements.
-- Provide insights on real-time news and announcements impacting the crypto market.
-### 3. Interaction Guidelines
-- Respond promptly and accurately to user queries.
-- Suggest safe and ethical trading practices.
-- Always remind users to do their own research (DYOR) and consult financial professionals where appropriate.
-### 4. Disclaimer
-- Remind users that cryptocurrency trading involves significant risk and past performance does not guarantee future results.
-- Clearly state that your responses are for informational purposes only and not financial advice.
-### Example Interactions
-#### Example 1: Market Analysis
-_User Query:_ "What’s the current trend of Bitcoin?"
-_Response:_ "Bitcoin is currently trading at $X, showing a [bullish/bearish] trend over the past 24 hours. Trading volume has [increased/decreased] by X%, and RSI indicates [overbought/oversold] conditions. Short-term support is at $Y, and resistance is at $Z."
-#### Example 2: Portfolio Review
-_User Query:_ "Is my portfolio balanced?"
-_Response:_ "Your portfolio comprises X% Bitcoin, Y% Ethereum, and Z% altcoins. To reduce risk, consider allocating X% to stablecoins or large-cap cryptocurrencies. Currently, your exposure to high-volatility assets is X%, which may pose additional risk."
-#### Example 3: Risk Management
-_User Query:_ "How do I protect my trades?"
-_Response:_ "You can use stop-loss orders at $X to limit potential losses or take-profit orders at $Y to secure gains. Avoid over-leveraging and limit each trade to a percentage of your total capital, such as 1-2%."
+You are a highly knowledgeable and reliable Crypto Trading Advisor and Analyzer. 
+Your goal is to assist users in understanding, analyzing, and making informed decisions about cryptocurrency trading. 
+You provide accurate, concise, and actionable advice based on real-time data, historical trends, and established best practices.
 """
 
-# Function to handle chatbot responses
-def respond(
-    message,
-    history: list[tuple[str, str]],
-    max_tokens,
-    temperature,
-    top_p,
-    frequency_penalty,
-    seed
-):
-    print(f"Received message: {message}")
-    print(f"History: {history}")
-
-    # Convert seed to None if -1 (meaning random)
-    if seed == -1:
-        seed = None
-
-    messages = [{"role": "system", "content": SYSTEM_PROMPT}]
-    print("System prompt added to messages.")
-
-    # Add conversation history to the context
-    for val in history:
-        user_part = val[0]
-        assistant_part = val[1]
-        if user_part:
-            messages.append({"role": "user", "content": user_part})
-        if assistant_part:
-            messages.append({"role": "assistant", "content": assistant_part})
-
-    # Append the latest user message
-    messages.append({"role": "user", "content": message})
-
-    # Start response generation
-    response = ""
-    print("Sending request to OpenAI API.")
-
-    for message_chunk in client.chat.completions.create(
-        model="meta-llama/Llama-3.3-70B-Instruct",
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-        frequency_penalty=frequency_penalty,
-        seed=seed,
-        messages=messages,
-    ):
-        token_text = message_chunk.choices[0].delta.content
-        response += token_text
+# CrewAI Integration
+llm = client  # Using the OpenAI client for CrewAI agents
+
+def run_crypto_crew(topic):
+    researcher = Agent(
+        role='Market Researcher',
+        goal=f'Uncover emerging trends and investment opportunities in the cryptocurrency market. Focus on the topic: {topic}.',
+        backstory='Identify groundbreaking trends and actionable insights.',
+        verbose=True,
+        tools=[search_tool],
+        allow_delegation=False,
+        llm=llm,
+        max_iter=3,
+        max_rpm=10,
+    )
+
+    analyst = Agent(
+        role='Investment Analyst',
+        goal=f'Analyze cryptocurrency market data to extract actionable insights. Focus on the topic: {topic}.',
+        backstory='Draw meaningful conclusions from cryptocurrency market data.',
+        verbose=True,
+        allow_delegation=False,
+        llm=llm,
+    )
+
+    research_task = Task(
+        description=f'Explore the internet to identify trends and investment opportunities. Topic: {topic}.',
+        agent=researcher,
+        expected_output='Detailed summary of research results.'
+    )
+
+    analyst_task = Task(
+        description=f'Analyze the market data to compile a concise report. Topic: {topic}.',
+        agent=analyst,
+        expected_output='Finalized version of the analysis report.'
+    )
+
+    crypto_crew = Crew(
+        agents=[researcher, analyst],
+        tasks=[research_task, analyst_task],
+        process=Process.sequential
+    )
+
+    result = crypto_crew.kickoff()
+    return result.raw
+
+# Chatbot Response Function
+def respond(message, history):
+    max_tokens = 512
+    temperature = 0.3
+    top_p = 0.95
+    frequency_penalty = 0.0
+    seed = None
+
+    if "analyze" in message.lower() or "trend" in message.lower():
+        response = run_crypto_crew(message)
         yield response
+    else:
+        messages = [{"role": "system", "content": SYSTEM_PROMPT}]
+        for user_part, assistant_part in history:
+            if user_part:
+                messages.append({"role": "user", "content": user_part})
+            if assistant_part:
+                messages.append({"role": "assistant", "content": assistant_part})
+        messages.append({"role": "user", "content": message})
 
-    print("Completed response generation.")
+        response = ""
+        for message_chunk in client.chat.completions.create(
+            model="meta-llama/Llama-3.3-70B-Instruct",
+            max_tokens=max_tokens,
+            stream=True,
+            temperature=temperature,
+            top_p=top_p,
+            frequency_penalty=frequency_penalty,
+            seed=seed,
+            messages=messages,
+        ):
+            token_text = message_chunk.choices[0].delta.content
+            response += token_text
+            yield response
 
 # Gradio UI
-chatbot = gr.Chatbot(height=600, show_copy_button=True, placeholder="Ask about crypto trading or analysis.", likeable=True)
-
-max_tokens_slider = gr.Slider(
-    minimum=1,
-    maximum=4096,
-    value=512,
-    step=1,
-    label="Max new tokens"
-)
-temperature_slider = gr.Slider(
-    minimum=0.1,
-    maximum=4.0,
-    value=0.7,
-    step=0.1,
-    label="Temperature"
-)
-top_p_slider = gr.Slider(
-    minimum=0.1,
-    maximum=1.0,
-    value=0.95,
-    step=0.05,
-    label="Top-P"
-)
-frequency_penalty_slider = gr.Slider(
-    minimum=-2.0,
-    maximum=2.0,
-    value=0.0,
-    step=0.1,
-    label="Frequency Penalty"
-)
-seed_slider = gr.Slider(
-    minimum=-1,
-    maximum=65535,
-    value=-1,
-    step=1,
-    label="Seed (-1 for random)"
-)
+chatbot = gr.Chatbot(height=600, show_copy_button=True, placeholder="Ask about crypto trading or analysis.")
 
 demo = gr.ChatInterface(
     fn=respond,
-    additional_inputs=[
-        max_tokens_slider,
-        temperature_slider,
-        top_p_slider,
-        frequency_penalty_slider,
-        seed_slider,
-    ],
     fill_height=True,
     chatbot=chatbot,
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()