Update app.py

app.py

@@ -238,7 +238,7 @@ if uploaded_file:
             if len(parts) > 1:
                 filtered_output = parts[1].strip()
             else:
-                filtered_output = full_output  
+                filtered_output = response 
             
         st.markdown("#### 💡 AI-based Medicine Analysis")
         st.text_area("LLM Output", filtered_output, height=300)
@@ -246,18 +246,18 @@ if uploaded_file:
         # Save txt and image
         txt_path = "medicine_analysis.txt"
         with open(txt_path, "w") as f:
-            f.write(response)
+            f.write(filtered_output)
 
         img_path = "medicine_analysis.png"
-        save_text_as_image(response, img_path)
+        save_text_as_image(filtered_output, img_path)
 
         st.markdown("#### 📥 Download (English)")
         col1, col2 = st.columns(2)
         with col1:
-            st.download_button("⬇️ English TXT", data=response.encode(), file_name="medicine_analysis.txt")
+            st.download_button("⬇️ English TXT", data=filtered_output.encode(), file_name="medicine_analysis.txt")
         with col2:
             with open(img_path, "rb") as img_file:
-                st.download_button("🖼️ English Image", data=img_file, file_name="medicine_analysis.png", mime="image/png")\
+                st.download_button("🖼️ English Image", data=img_file, file_name="medicine_analysis.png", mime="image/png")
                 
     # if response and st.button("🌐 Translate to Hindi"):
     #     with st.spinner("Translating to Hindi..."):