app.py

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

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

def ImageChat(image):
    # 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}"

    # Custom prompt
    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.

    Key Elements to Include:

    **Trend Analysis:** Identify the current trend and its strength (e.g., bullish, bearish, consolidation).
    **Price Action:** Describe recent price movements and patterns (e.g., support and resistance levels, candlestick patterns).
    **Entry and Exit Points:** Specify precise entry and exit points based on the chart’s data.
    **Hold Conditions:** Determine the conditions under which holding the position is advisable.
    **Risk Management:** Outline stop-loss levels, position sizing, and other risk management techniques.
    **Timeframe:** Indicate the expected duration of the trade in terms of the chart’s timesteps, balancing profit potential with trade duration.
    **Profit Potential:** Estimate the potential profit in absolute values as shown in the chart.
    Example Analysis Structure:

    **Trend Analysis:** The stock is currently in a strong bullish trend, evidenced by higher highs and higher lows over the past **X** timesteps.
    **Price Action:** Recent price action shows a breakout from a key resistance level at **$Y**, indicating strong upward momentum.
    **Entry Point:** Enter the trade at **$Z**, where the breakout is confirmed.
    **Exit Point:** Exit the trade at **$A**, near the next major resistance level, for a potential profit of **$B** per share.
    **Hold Conditions:** Hold the position as long as the price remains above the support level at **$C**.
    **Risk Management:** 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.
    **Timeframe:** The expected duration of the trade is **F** timesteps, aiming for a profit of **$G** in this period.
    """

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

    # 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."

        # Apply rich formatting to the response
        formatted_response = response.text
        for title in ["Trend Analysis", "Price Action", "Entry Point", "Exit Point", "Hold Conditions", "Risk Management", "Timeframe", "Profit Potential"]:
            formatted_response = formatted_response.replace(title, f"**{title.upper()}**")
        formatted_response = re.sub(r"(\d+\.?\d*)", r"**\1**", formatted_response)
        formatted_response = formatted_response.replace('\n', '\n\n')
        return formatted_response
    except ValueError as e:
        return f"Error in generating response: {e}"

# Define the Gradio interface
with gr.Blocks() as app:
    gr.Markdown("# Image Chat")
    image_input = gr.Image(label="Image")
    analyze_button = gr.Button("Analyze", elem_id="analyze_button")
    prompt_input = gr.Textbox(label="Prompt", value="Analyze the attached stock chart image as a technical quant analyst...")
    response_output = gr.Textbox(label="Response")
    
    def analyze_image(image):
        return ImageChat(image)
    
    analyze_button.click(fn=analyze_image, inputs=image_input, outputs=response_output)

# Style the analyze button
app.css = """
#analyze_button {
    background-color: #26de81;
    color: #ffffff;
}
"""

app.launch()