Update app.py

app.py

@@ -17,7 +17,6 @@ logging.basicConfig(filename='debug.log', level=logging.DEBUG, format='%(asctime
 processor = AutoProcessor.from_pretrained("mobenta/chart_analysis")
 model = AutoModelForPreTraining.from_pretrained("mobenta/chart_analysis")
 
-@spaces.GPU
 def predict(image, input_text):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     model.to(device)
@@ -132,47 +131,23 @@ def create_stock_chart(data, ticker, filename='chart.png', timeframe='1d', indic
         resized_image = image.resize(new_size, Image.LANCZOS)
         resized_image.save(filename)
 
-        logging.debug(f"Resized image with timeframe {timeframe} and ticker {ticker} saved to {filename}")
+        logging.debug(f"Resized image saved to {filename}")
+        return filename
     except Exception as e:
-        logging.error(f"Error creating or resizing chart: {e}")
-        raise
-
-def combine_images(image_paths, output_path='combined_chart.png'):
-    try:
-        logging.debug(f"Combining images {image_paths} into {output_path}")
-        images = [Image.open(path) for path in image_paths]
-        
-        # Calculate total width and max height for combined image
-        total_width = sum(img.width for img in images)
-        max_height = max(img.height for img in images)
-
-        combined_image = Image.new('RGB', (total_width, max_height))
-        x_offset = 0
-        for img in images:
-            combined_image.paste(img, (x_offset, 0))
-            x_offset += img.width
-
-        combined_image.save(output_path)
-        logging.debug(f"Combined image saved to {output_path}")
-        return output_path
-    except Exception as e:
-        logging.error(f"Error combining images: {e}")
+        logging.error(f"Error creating stock chart: {e}")
         raise
 
 def gradio_interface(ticker1, ticker2, ticker3, ticker4, start_date, end_date, query, analysis_type, interval, indicators):
     try:
-        logging.debug(f"Starting gradio_interface with tickers: {ticker1}, {ticker2}, {ticker3}, {ticker4}, start_date: {start_date}, end_date: {end_date}, query: {query}, analysis_type: {analysis_type}, interval: {interval}")
-
-        tickers = [ticker1, ticker2, ticker3, ticker4]
         chart_paths = []
+        tickers = [ticker1, ticker2, ticker3, ticker4]
 
-        for i, ticker in enumerate(tickers):
-            if ticker:
-                data = fetch_stock_data(ticker, start=start_date, end=end_date, interval=interval)
-                with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as temp_chart:
-                    chart_path = temp_chart.name
-                    create_stock_chart(data, ticker, chart_path, timeframe=interval, indicators=indicators)
-                    chart_paths.append(chart_path)
+        for ticker in tickers:
+            if ticker.strip():
+                data = fetch_stock_data(ticker, start_date, end_date, interval)
+                chart_path = f"{ticker}_chart.png"
+                create_stock_chart(data, ticker, filename=chart_path, timeframe=interval, indicators=indicators)
+                chart_paths.append(chart_path)
 
         if analysis_type == 'Comparative Analysis' and len(chart_paths) > 1:
             with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as temp_combined_chart:
@@ -181,7 +156,6 @@ def gradio_interface(ticker1, ticker2, ticker3, ticker4, start_date, end_date, q
                 insights = predict(Image.open(combined_chart_path), query)
                 return insights, combined_chart_path
 
-        # No comparative analysis, just return the single chart
         if chart_paths:
             insights = predict(Image.open(chart_paths[0]), query)
             return insights, chart_paths[0]
@@ -224,4 +198,4 @@ def gradio_app():
     demo.launch()
 
 if __name__ == "__main__":
-    gradio_app()
+    gradio_app()