app.py

import google.generativeai as genai
import gradio as gr
import numpy as np
import PIL.Image
import io

genai.configure(api_key="AIzaSyAj-b3sO_wUguMdpXWScxKzMHxb8C5cels")

# Define the questions for different chart types
questions = {
    "Forex": [
        "Is the current trend bullish or bearish for this pair?",
        "What are the key support and resistance levels?",
        "Are there any significant economic events impacting this pair?",
        "What do candlestick patterns suggest about future price movements?",
        "How do moving averages influence this pair's trend?",
        "What does the RSI indicate about this pair’s strength?",
        "Are there any MACD signals indicating a change in trend?",
        "How does volume affect the current forex pair's trend?",
        "Is there a potential breakout or breakdown in the chart?",
        "What are the Fibonacci retracement levels for this pair?"
    ],
    "Crypto": [
        "What is the current trend of this cryptocurrency?",
        "Are there any significant candlestick patterns forming?",
        "How do moving averages impact the crypto's price action?",
        "What are the support and resistance levels for this crypto?",
        "What does the RSI indicate about the crypto's market condition?",
        "Are there any divergences between price and MACD?",
        "How does the volume confirm the current trend?",
        "Is there a potential reversal signal in the chart?",
        "What are the potential breakout points in the chart?",
        "How do news or events affect the crypto's price?"
    ],
    "Stocks": [
        "Is the stock currently in an uptrend or downtrend?",
        "What are the key support and resistance levels?",
        "Are there any significant candlestick patterns forming?",
        "How do moving averages influence the stock's price action?",
        "What does the RSI indicate about the stock's market condition?",
        "Are there any divergence signals between price and MACD?",
        "How does the volume confirm the current trend?",
        "Is there a potential reversal signal in the stock's chart?",
        "What are the potential breakout points in the stock's chart?",
        "How do earnings reports or company news impact the stock's price?"
    ],
    # Add similar lists for Futures and Options
}

def ImageChat(image, chart_type, ticker, prompt="Analyze", analyze_more=None):
    # Check image file and convert to a PIL Image object
    if isinstance(image, np.ndarray):
        img = PIL.Image.fromarray(image)
    else:
        try:
            img = PIL.Image.open(image)
        except (AttributeError, IOError) as e:
            return f"Invalid image provided. Please provide a valid image file. Error: {e}"

    # Load model
    model = genai.GenerativeModel("gemini-pro-vision")

    # Determine the prompt to use
    if analyze_more:
        custom_prompt = f"Act as a genius Trader and answer the following request: {analyze_more} after analyzing the chart thoroughly."
    elif prompt == "Analyze":
        custom_prompt = (
            "Analyze the attached stock chart image as a technical quant analyst. Based on the trend, movement, and price action visible in the chart, make intelligent and substantial trading suggestions. Provide clear entry, exit, and hold conditions, along with detailed risk management strategies. Discuss the expected timeframe for the trade, specifying the duration in terms of the timesteps shown in the chart. Aim for a good profit-to-time ratio, favoring high profits within shorter timeframes unless longer timeframes yield significantly higher profits. Reference absolute values from the chart rather than relying on general strategies. Avoid basic explanations of indicators, and focus on providing in-depth analysis.\n\n"

            "Key Elements to Include:\n\n"

            "Trend Analysis: Identify the current trend and its strength (e.g., bullish, bearish, consolidation).\n"
            "Price Action: Describe recent price movements and patterns (e.g., support and resistance levels, candlestick patterns).\n"
            "Entry and Exit Points: Specify precise entry and exit points based on the chart's data.\n"
            "Hold Conditions: Determine the conditions under which holding the position is advisable.\n"
            "Risk Management: Outline stop-loss levels, position sizing, and other risk management techniques.\n"
            "Timeframe: Indicate the expected duration of the trade in terms of the chart's timesteps, balancing profit potential with trade duration.\n"
            "Profit Potential: Estimate the potential profit in absolute values as shown in the chart.\n"
            "Candlestick Analysis: Spot and describe any notable candlestick formations.\n"
            "Other Indicator Analysis: Include analysis of additional technical indicators present on the chart.\n"
            "Additional Observations: Note any other important observations made from the chart.\n\n"

            "Example Analysis Structure:\n\n"

            "Trend Analysis:\n"
            "The stock is currently in a strong bullish trend, evidenced by higher highs and higher lows over the past X timesteps.\n\n"

            "Price Action:\n"
            "Recent price action shows a breakout from a key resistance level at $Y, indicating strong upward momentum.\n\n"

            "Entry Point:\n"
            "Enter the trade at $Z, where the breakout is confirmed.\n\n"

            "Exit Point:\n"
            "Exit the trade at $A, near the next major resistance level, for a potential profit of $B per share.\n\n"

            "Hold Conditions:\n"
            "Hold the position as long as the price remains above the support level at $C.\n\n"

            "Risk Management:\n"
            "Set a stop-loss at $D to limit potential losses to $E per share, and use position sizing to ensure the total risk is within acceptable limits.\n\n"

            "Timeframe:\n"
            "The expected duration of the trade is F timesteps, aiming for a profit of $G in this period.\n\n"

            "Candlestick Analysis:\n"
            "Spot and describe any significant candlestick patterns, such as doji, hammer, or engulfing patterns, and their implications for the trade.\n\n"

            "Other Indicator Analysis:\n"
            "Evaluate additional technical indicators present on the chart, such as moving averages, RSI, MACD, or Bollinger Bands, and discuss how they support the trading strategy.\n\n"

            "Additional Observations:\n"
            "Note any other relevant observations, such as volume spikes, unusual price gaps, or divergence patterns, that could impact the trade's success.\n\n"

            "By including these detailed elements, the analysis will provide a comprehensive and actionable trading strategy based on the stock chart's data."
        )
    else:
        custom_prompt = f"Act as a genius Trader and answer the following request: {prompt} after analyzing the chart thoroughly."

    # Generate response
    try:
        response = model.generate_content([custom_prompt, img])
        if not response or not response.text:
            return "No valid response received. The response might have been blocked."
        
        # Formatting the response
        formatted_response = ""
        for line in response.text.split("\n"):
            if line.strip().endswith(":"):
                formatted_response += f"**{line.strip()}**\n"
            else:
                formatted_response += line + "\n"
        
        return formatted_response
    except ValueError as e:
        return f"Error in generating response: {e}"

def populate_questions(chart_type):
    return questions.get(chart_type, [])

def update_analyze_more(chart_type):
    return gr.update(choices=populate_questions(chart_type))

app = gr.Interface(
    fn=ImageChat,
    inputs=[
        gr.Image(label="Image"), 
        gr.Dropdown(label="Chart Type", choices=["Forex", "Crypto", "Stocks"], type="value", value="Stocks"),
        gr.Textbox(label="Ticker", placeholder="Enter ticker symbol"),
        gr.Textbox(label="Prompt", value="Analyze"),
        gr.Dropdown(label="Analyze more", choices=[], type="value", interactive=True)
    ],
    outputs=gr.Textbox(label="Response"),
    title="Trading View Chat Analyzer",
    theme=gr.themes.Soft()
)

# Create a function to dynamically update the dropdown based on the selected chart type
def dynamic_analyze_more(chart_type):
    questions_list = populate_questions(chart_type)
    return gr.update(choices=questions_list)

# Create a new Gradio Block to handle interactions
with gr.Blocks() as demo:
    chart_type = gr.Dropdown(label="Chart Type", choices=["Forex", "Crypto", "Stocks"], type="value", value="Stocks")
    ticker = gr.Textbox(label="Ticker", placeholder="Enter ticker symbol")
    prompt = gr.Textbox(label="Prompt", value="Analyze")
    analyze_more = gr.Dropdown(label="Analyze more", choices=[], type="value", interactive=True)
    image = gr.Image(label="Image")
    response = gr.Textbox(label="Response")

    chart_type.change(dynamic_analyze_more, inputs=chart_type, outputs=analyze_more)
    
    gr.Interface(
        fn=ImageChat,
        inputs=[image, chart_type, ticker, prompt, analyze_more],
        outputs=response
    ).launch()

demo.launch()