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 BytesIO
+from io import StringIO
 import google.generativeai as genai
 
 # Setup
@@ -66,9 +66,6 @@ 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()
@@ -88,17 +85,8 @@ if uploaded_file and question:
                     {"text": question}
                 ]
                 response = model.generate_content(contents=content)
-                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)
+                st.success("💡 Answer:")
+                st.write(response.text)
 
             except Exception as e:
                 st.error(f"Error processing image: {e}")
@@ -119,17 +107,7 @@ if uploaded_file and question:
                 try:
                     prompt = f"Context: {extracted_text}\n\nQuestion: {question}\n\nAnswer:"
                     response = model.generate_content(prompt)
-                    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)
-
+                    st.success("💡 Answer:")
+                    st.write(response.text)
                 except Exception as e:
                     st.error(f"Error generating answer: {e}")