Update app.py

app.py

@@ -6,7 +6,7 @@ import openpyxl
 from pdfminer.high_level import extract_text
 import csv
 from pptx import Presentation
-from io import StringIO
+from io import BytesIO
 import google.generativeai as genai
 
 # Setup
@@ -66,6 +66,9 @@ uploaded_file = st.file_uploader("Upload a document or image", type=["docx", "xl
 # Question input
 question = st.text_input("💬 Enter your question:")
 
+# Output type selection
+output_type = st.radio("Choose output type", ("Text", "Audio", "Image", "Both"))
+
 # Processing
 if uploaded_file and question:
     file_ext = uploaded_file.name.split('.')[-1].lower()
@@ -85,8 +88,17 @@ if uploaded_file and question:
                     {"text": question}
                 ]
                 response = model.generate_content(contents=content)
-                st.success("💡 Answer:")
-                st.write(response.text)
+                if output_type in ["Text", "Both"]:
+                    st.success("💡 Answer:")
+                    st.write(response.text)
+                if output_type in ["Audio", "Both"]:
+                    audio_response = model.generate_audio(response.text)
+                    audio_data = base64.b64decode(audio_response.audio_data)
+                    st.audio(audio_data, format="audio/wav")
+                if output_type in ["Image", "Both"]:
+                    image_response = model.generate_image(response.text)
+                    image_data = base64.b64decode(image_response.image_data)
+                    st.image(image_data, caption="Generated Image", use_column_width=True)
 
             except Exception as e:
                 st.error(f"Error processing image: {e}")
@@ -107,7 +119,17 @@ if uploaded_file and question:
                 try:
                     prompt = f"Context: {extracted_text}\n\nQuestion: {question}\n\nAnswer:"
                     response = model.generate_content(prompt)
-                    st.success("💡 Answer:")
-                    st.write(response.text)
+                    if output_type in ["Text", "Both"]:
+                        st.success("💡 Answer:")
+                        st.write(response.text)
+                    if output_type in ["Audio", "Both"]:
+                        audio_response = model.generate_audio(response.text)
+                        audio_data = base64.b64decode(audio_response.audio_data)
+                        st.audio(audio_data, format="audio/wav")
+                    if output_type in ["Image", "Both"]:
+                        image_response = model.generate_image(response.text)
+                        image_data = base64.b64decode(image_response.image_data)
+                        st.image(image_data, caption="Generated Image", use_column_width=True)
+
                 except Exception as e:
-                    st.error(f"Error generating answer: {e}")
+                    st.error(f"Error generating answer: {e}")