Update app.py

app.py

@@ -4,12 +4,8 @@ import logging
 from sklearn.feature_extraction.text import TfidfVectorizer
 from sklearn.metrics.pairwise import cosine_similarity
 from transformers import MarianMTModel, MarianTokenizer
-from PIL import Image, ImageDraw, ImageFont
-import textwrap
-import os
 from gtts import gTTS
 import tempfile
-from diffusers import DiffusionPipeline
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -135,19 +131,6 @@ class MultilingualGitaAnalyzer:
             logger.error(f"Error in semantic search: {e}")
             return ["An error occurred during semantic search."]
 
-    def generate_image(self, text):
-        """Generate an image using the DiffusionPipeline."""
-        try:
-            pipe = DiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
-            prompt = f"in the style of CNSTLL, {text}, night time, cinestill 800T"
-            image = pipe(prompt).images[0]
-            img_path = 'generated_image.png'
-            image.save(img_path)
-            return img_path
-        except Exception as e:
-            logger.error(f"Error generating image: {e}")
-            return None
-
     def generate_audio(self, text):
         """Generate audio from the given text."""
         try:
@@ -183,22 +166,16 @@ class MultilingualGitaAnalyzer:
                 explanation = "In simple terms, duty means doing what you are supposed to do. It's like your responsibilities or tasks that you need to complete. In today's world, it could be your job, taking care of your family, or helping others. It's about doing the right thing, even when it's hard."
                 translated_explanation = self.translate_from_english(explanation, target_language=preferred_language)
 
-                # Generate an image for the first result
-                if translated_results:
-                    image_path = self.generate_image(translated_results[0])
-                else:
-                    image_path = None
+                # Combine results into a story format
+                story = "\n\n".join(translated_results) + "\n\n" + translated_explanation
 
-                # Generate audio for the first result
-                if translated_results:
-                    audio_path = self.generate_audio(translated_results[0])
-                else:
-                    audio_path = None
+                # Generate audio for the story
+                audio_path = self.generate_audio(story)
 
-                return "\n\n".join(translated_results) + "\n\n" + translated_explanation, image_path, audio_path
+                return story, audio_path
             except Exception as e:
                 logger.error(f"Error processing query: {e}")
-                return "An unexpected error occurred.", None, None
+                return "An unexpected error occurred.", None
 
         def daily_quote():
             # Placeholder for daily quote logic
@@ -229,11 +206,10 @@ class MultilingualGitaAnalyzer:
                     submit_btn = gr.Button("Submit")
 
                 with gr.Column():
-                    output = gr.Textbox(label="Vedas Wisdom")
-                    image_output = gr.Image(label="Relevant Image")
+                    output = gr.Textbox(label="Vedas Wisdom Story")
                     audio_output = gr.Audio(label="Relevant Audio")
 
-            submit_btn.click(process_query, inputs=[question, preferred_language], outputs=[output, image_output, audio_output])
+            submit_btn.click(process_query, inputs=[question, preferred_language], outputs=[output, audio_output])
 
             gr.Markdown("## Daily Quote")
             daily_quote_output = gr.Textbox(label="Daily Quote", value=daily_quote())