Update app.py

app.py

@@ -13,34 +13,38 @@ client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
 
 # Define function to process image description and generate audio
 def process_image_and_generate_audio(image_url):
-    response = client.chat.completions.create(
-        model="gpt-4o",
-        messages=[
-            {
-                "role": "user",
-                "content": [
-                    {"type": "text", "text": "Explain every single thing about this image"},
-                    {
-                        "type": "image_url",
-                        "image_url": {"url": image_url},
-                    },
-                ],
-            }
-        ],
-        max_tokens=300,
-    )
+    try:
+        response = client.chat.completions.create(
+            model="gpt-4o",
+            messages=[
+                {
+                    "role": "user",
+                    "content": [
+                        {"type": "text", "text": "Explain every single thing about this image"},
+                        {
+                            "type": "image_url",
+                            "image_url": {"url": image_url},
+                        },
+                    ],
+                }
+            ],
+            max_tokens=300,
+        )
 
-    # Get content from response
-    content = response.choices[0].message.content
+        # Get content from response
+        content = response.choices[0].message.content
 
-    # Generate audio from content
-    audio_response = client.audio.speech.create(
-        model="tts-1",
-        voice="alloy",
-        input=content,
-    )
+        # Generate audio from content
+        audio_response = client.audio.speech.create(
+            model="tts-1",
+            voice="alloy",
+            input=content,
+        )
 
-    return content, audio_response
+        return content, audio_response
+    except Exception as e:
+        st.error(f"An error occurred: {str(e)}")
+        return None, None
 
 # Streamlit UI
 def main():
@@ -56,16 +60,17 @@ def main():
             # Generate content and audio
             content, audio_response = process_image_and_generate_audio(image_url)
 
-            # Write audio to a temporary file
-            with tempfile.NamedTemporaryFile(suffix=".mp3", delete=False) as f:
-                audio_response.stream_to_file(f.name)
+            if content is not None and audio_response is not None:
+                # Write audio to a temporary file
+                with tempfile.NamedTemporaryFile(suffix=".mp3", delete=False) as f:
+                    audio_response.stream_to_file(f.name)
 
-            # Display content
-            st.markdown("**Description:**")
-            st.write(content)
+                # Display content
+                st.markdown("**Description:**")
+                st.write(content)
 
-            # Display the audio
-            st.audio(open(f.name, "rb").read(), format="audio/mp3")
+                # Display the audio
+                st.audio(open(f.name, "rb").read(), format="audio/mp3")
 
 if __name__ == "__main__":
     main()