Update synthscope.py

synthscope.py

@@ -14,25 +14,51 @@ google_search_tool = Tool(
 
 model_id = "gemini-2.0-flash"
 
-def GoogleSearchImageGen(prompt, image_style):
-    # Define style-specific prompts
+def GoogleSearchImageGen(prompt, image_style, voice):
+    def wave_file(filename, pcm, channels=1, rate=24000, sample_width=2):
+        with wave.open(filename, "wb") as wf:
+            wf.setnchannels(channels)
+            wf.setsampwidth(sample_width)
+            wf.setframerate(rate)
+            wf.writeframes(pcm)
+
+    # define style-specific prompts
     style_prompts = {
         "Comic": f"{prompt}. Convert the search result into a well-crafted text-to-image prompt that generates a comic book-style image.",
         "Cartoon": f"{prompt}. Convert the search result into a well-crafted text-to-image prompt that generates a cartoon-style image.",
         "3D": f"{prompt}. Convert the search result into a well-crafted text-to-image prompt that generates a Pixar-style 3D image.",
         "Anime": f"{prompt}. Convert the search result into a well-crafted text-to-image prompt that generates an Anime-style image.",
         "Ghibli": f"{prompt}. Convert the search result into a well-crafted text-to-image prompt that generates a Ghibli-style image.",
-        "Victorian": f"{prompt}. Convert the search result into a well-crafted text-to-image prompt that generates a Victorian-era image."
+        "Victorian": f"{prompt}. Convert the search result into a well-crafted text-to-image prompt that generates a Victorian-era image.",
+        "Movie": f"{prompt}. Convert the search result into a well-crafted text-to-image prompt that generates a Movie-style image.",
+        "Star Wars": f"{prompt}. Convert the search result into a well-crafted text-to-image prompt that generates a Star Wars-style image.",
+        "Marvel": f"{prompt}. Convert the search result into a well-crafted text-to-image prompt that generates a Marvel-style image."
+    }
+
+    # define audio output voice
+    select_voice = {
+        "Kore": "Kore",
+        "Zephyr": "Zephyr",
+        "Orus": "Orus",
+        "Fenrir": "Fenrir",
+        "Charon": "Charon",
+        "Umbriel": "Umbriel",
+        "Schedar": "Schedar",
+        "Iapetus": "Iapetus",
+        "Puck": "Puck",
+        "Gacrux": "Gacrux"
     }
-    
+
     # Check if the image_style is supported
     if image_style in style_prompts:
-      image_gen_prompt = style_prompts[image_style]
-    
+        image_gen_prompt = style_prompts[image_style]
+    else:
+        return "Invalid image style", None, None
+
     # Initialize variables to avoid UnboundLocalError
     search_result = ""
     image = None
-    
+
     try:
         # Get search result to be displayed to the user
         response = client.models.generate_content(
@@ -49,6 +75,28 @@ def GoogleSearchImageGen(prompt, image_style):
             if each.text:
                 search_result += each.text
 
+        # Generate audio from search result
+        audio_resp = client.models.generate_content(
+            model="gemini-2.5-flash-preview-tts",
+            contents=search_result,
+            config=types.GenerateContentConfig(
+                response_modalities=["AUDIO"],
+                speech_config=types.SpeechConfig(
+                    voice_config=types.VoiceConfig(
+                        prebuilt_voice_config=types.PrebuiltVoiceConfig(
+                            voice_name=select_voice[voice],
+
+                        )
+                    )
+                ),
+            )
+        )
+
+        data = audio_resp.candidates[0].content.parts[0].inline_data.data
+
+        audio_output_file = "out.wav"
+        wave_file(audio_output_file, data)
+
         # Generate image prompt from search result
         output = client.models.generate_content(
             model=model_id,
@@ -85,6 +133,6 @@ def GoogleSearchImageGen(prompt, image_style):
     except Exception as e:
         print(f"Error occurred: {e}")
         # Return default values in case of error
-        return search_result or "No search result available", None
+        return search_result or "No search result available", None, None
 
-    return search_result, image
+    return search_result, image, audio_output_file