support convert json to excel and download

app.py

@@ -112,6 +112,22 @@ if clone_repo():
         st.write("## アップロードされたファイル")
         st.image(pil_image, caption="アップロードされたファイル", use_column_width=True)
     
+    if 'json_predictions' in st.session_state:
+        prev_json_predictions = st.session_state.json_predictions
+        prev_excel_file_path = st.session_state.excel_file_path
+        with col2:
+            st.write("## 結果")
+            # 提供下载链接
+            with open(prev_excel_file_path, "rb") as file:
+                st.download_button(
+                    label="Download Excel",
+                    data=file,
+                    file_name="output.xlsx",
+                    mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+                )
+            st.write("解析後の内容:")
+            st.json(prev_json_predictions)
+        
     if text_rec:
         with col2:
             st.write("## 結果")
@@ -126,9 +142,24 @@ if clone_repo():
             
                 # Simulate OCR result as a JSON object
                 json_predictions = ia.get_json_result(predictions)
+                st.session_state.json_predictions = json_predictions
+                
+                # Convert JSON to Excel
+                excel_file_path = "output.xlsx"
+                st.session_state.excel_file_path = excel_file_path
+                ia.json_to_excel_with_links(json_predictions, excel_file_path)
             
                 # After model finishes
                 status_placeholder.success('ファイルの解析が完了しました!')
+
+                # 提供下载链接
+                with open(excel_file_path, "rb") as file:
+                    st.download_button(
+                        label="Download Excel",
+                        data=file,
+                        file_name="output.xlsx",
+                        mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+                    )
             
                 # Display the result
                 st.write("解析後の内容:")

requirements.txt

@@ -1,3 +1,5 @@
 surya-ocr==0.4.15
 pypdfium2==4.30.0
-openai==1.35.13
+openai==1.35.13
+pandas==2.2.2
+openpyxl==3.1.5