error handling for plotting when fetching data goes wrong

app.py

@@ -52,14 +52,14 @@ with gr.Blocks() as demo:
     analyze_button.click(
             fn=plot_stock,
             inputs=[ticker],
-            outputs=plot_output, 
-            api_name="analyze_stock")
+            outputs=[plot_output],
+            api_name="plot_stock")
     md_output = gr.Markdown(label="Output Box")
     analyze_button.click(
             fn=ask_stock_agent,
             inputs=[agent_instance, ticker],
             outputs=md_output, 
-            api_name="analyze_stock")
+            api_name="ask_stock_agent")
 
 
 if __name__ == "__main__":

src/technical_analysis.py

@@ -52,7 +52,6 @@ class TechnicalAnalysis():
     def run(
             self
             ) -> None:
-        # fetch data from yf
         """
         Main entry point for the TechnicalAnalysis object.
         This method:
@@ -60,6 +59,7 @@ class TechnicalAnalysis():
         - computes the technical analysis metrics
         - plots the price and TA metrics.
         """
+        # fetch data from yf
         self.df = self.fetch_data()
         # add the features based on technical analysis
         if self.df.shape[0] > 0:
@@ -79,10 +79,13 @@ class TechnicalAnalysis():
             else:
                 fig = None
         else:
-            fig = None
+            if self.plot_ta:
+                fig = self.get_fetcherror_fig(message='failed fetching data')
+            else:
+                fig = None
 
         return self.df, fig
-
+    
     def fetch_data(
             self
             ) -> pd.DataFrame:
@@ -286,10 +289,39 @@ class TechnicalAnalysis():
             ax.fill_between(df.index, df['Low'], df['High'], color='gray', alpha=0.3)
         ax.grid(True, linestyle='--', alpha=0.7)
 
+    def get_fetcherror_fig(
+            self,
+            message
+            ) -> plt.Figure:
+        """
+        Fetches images/plot_error.png, annotates it and returns it as a matplotlib.pyplot.Figure object
+        Args:
+        message (str): message to be annotated on the displayed image
+        Returns:
+        plt.Figure: figure object containing the annotated image
+        """
+        
+        fig, ax = plt.subplots(
+            figsize=(5, 5)
+        )
+
+        # Load and display the image
+        parentdir = os.path.dirname(os.path.dirname(__file__)) 
+        fname = os.path.join(parentdir, 'images', 'plot_error.png')
+        img = plt.imread(fname)
+        ax.imshow(img)
+
+        # Remove axes ticks and labels
+        # ax.set_xticks([])
+        # ax.set_yticks([])
+        ax.axis('off')  # removes both ticks and axes lines completely
+
+        ax.text(0.5, 0.05, message, fontsize=20, ha='center', va='center', transform=ax.transAxes)
+        return fig
 
 if __name__ == '__main__':
     ticker = 'GOOG'
-    df = TechnicalAnalysis(ticker, debug=False).run()
+    df, fig = TechnicalAnalysis(ticker, plot_ta=True, savefig=False, debug=False).run()
     print(f'columns: {df.columns}')