Update app.py

app.py

@@ -4,11 +4,32 @@ import numpy as np
 import PIL.Image
 import io
 
-# Configure API key for Google Generative AI
 genai.configure(api_key="AIzaSyAj-b3sO_wUguMdpXWScxKzMHxb8C5cels")
 
-# Function to handle image input and generate analysis
-def ImageChat(image, user_prompt):
+# 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.
+"""
+
+def ImageChat(image):
     # Check image file and convert to a PIL Image object
     if isinstance(image, np.ndarray):
         img = PIL.Image.fromarray(image)
@@ -21,50 +42,23 @@ def ImageChat(image, user_prompt):
     # Load model
     model = genai.GenerativeModel("gemini-pro-vision")
 
-    # Define custom prompt for stock chart analysis
-    custom_prompt = f"""
-    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.
-
-    {user_prompt}
-    """
-
     # 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."
-        return response.text
+        # Add rich formatting to the output
+        formatted_response = response.text.replace("\n", "<br>").replace("$", "<b>$").replace("**", "</b>")
+        return formatted_response
     except ValueError as e:
         return f"Error in generating response: {e}"
 
-# Create the Gradio interface
 app = gr.Interface(
     fn=ImageChat,
-    inputs=[gr.Image(label="Stock Chart Image", type="pil"), gr.Textbox(label="Continue Chat")],
-    outputs=gr.Markdown(label="Analysis"),
+    inputs=gr.Image(label="Upload Stock Chart Image"),
+    outputs=gr.HTML(label="Analysis Response"),
     title="Stock Chart Analysis",
     theme=gr.themes.Soft()
 )
 
-# Launch the app
 app.launch()