Update app.py

app.py

@@ -2,11 +2,11 @@ import google.generativeai as genai
 import gradio as gr
 import numpy as np
 import PIL.Image
-import re
+import io
 
 genai.configure(api_key="AIzaSyAj-b3sO_wUguMdpXWScxKzMHxb8C5cels")
 
-def ImageChat(image):
+def ImageChat(image, prompt="Analyze"):
     # Check image file and convert to a PIL Image object
     if isinstance(image, np.ndarray):
         img = PIL.Image.fromarray(image)
@@ -16,68 +16,60 @@ def ImageChat(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")
 
+    # 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.\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\n"
+        "Example Analysis Structure:\n\n"
+        "**Trend Analysis:** The stock is currently in a strong bullish trend, evidenced by higher highs and higher lows over the past **X** timesteps. 📈\n"
+        "**Price Action:** Recent price action shows a breakout from a key resistance level at **$Y**, indicating strong upward momentum. 💹\n"
+        "**Entry Point:** Enter the trade at **$Z**, where the breakout is confirmed. 🔍\n"
+        "**Exit Point:** Exit the trade at **$A**, near the next major resistance level, for a potential profit of **$B** per share. 💵\n"
+        "**Hold Conditions:** Hold the position as long as the price remains above the support level at **$C**.\n"
+        "**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. 🚨\n"
+        "**Timeframe:** The expected duration of the trade is **F** timesteps, aiming for a profit of **$G** in this period. ⏳\n"
+    )
+
     # 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')
+        
+        # Formatting the response
+        formatted_response = ""
+        for line in response.text.split("\n"):
+            if line.strip().endswith(":"):
+                formatted_response += f"**{line.strip().upper()}**\n"
+            else:
+                formatted_response += line.replace("$", "**$").replace(" ", "** ").replace("** ", "**")
+        
         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 = gr.Interface(
+    fn=ImageChat,
+    inputs=[gr.Image(label="Image"), gr.Text(label="Prompt", default="Analyze")],
+    outputs=gr.Text(label="Response"),
+    title="Trading View Chat Analyzer",
+    theme=gr.themes.Soft()
+)
 
 app.launch()