Upload app.py with huggingface_hub

app.py

@@ -5,9 +5,10 @@ from transformers import LlavaForConditionalGeneration, AutoProcessor
 from PIL import Image
 import gc
 import time
-import re
+import gc
 import os
 import shutil
+import json
 from pathlib import Path
 
 # Storage optimization - redirect cache to temporary directories
@@ -323,6 +324,56 @@ def answer_question(image, question):
     
     return result if result else "❌ No answer generated"
 
+def export_joycaption_data(keywords, custom_instructions, question, engaging_caption, casual_caption, keywords_caption, qa_answer):
+    """Export all JoyCaption data as downloadable JSON"""
+    try:
+        # Collect all the data
+        data = {
+            "timestamp": time.strftime("%Y-%m-%d %H:%M:%S"),
+            "source": "JoyCaption",
+            "data": {}
+        }
+        
+        # Add input fields
+        if keywords and keywords.strip():
+            data["data"]["keywords"] = keywords.strip()
+            
+        if custom_instructions and custom_instructions.strip():
+            data["data"]["custom_instructions"] = custom_instructions.strip()
+            
+        if question and question.strip():
+            data["data"]["question"] = question.strip()
+        
+        # Add generated captions
+        if engaging_caption and engaging_caption.strip():
+            data["data"]["caption_engaging"] = engaging_caption.strip()
+            
+        if casual_caption and casual_caption.strip():
+            data["data"]["caption_casual_friend"] = casual_caption.strip()
+            
+        if keywords_caption and keywords_caption.strip():
+            data["data"]["caption_keywords"] = keywords_caption.strip()
+            
+        if qa_answer and qa_answer.strip():
+            data["data"]["qa_answer"] = qa_answer.strip()
+        
+        # Check if we have any data to export
+        if not data["data"]:
+            return "❌ No data to export. Generate some captions first!", None
+        
+        # Create JSON string
+        json_string = json.dumps(data, indent=2, ensure_ascii=False)
+        
+        # Create filename with timestamp
+        filename = f"joycaption_data_{time.strftime('%Y%m%d_%H%M%S')}.json"
+        
+        # Return success message and file data
+        fields_count = len(data["data"])
+        return f"✅ Exported {fields_count} fields: {', '.join(data['data'].keys())}", (json_string, filename)
+        
+    except Exception as e:
+        return f"❌ Export failed: {str(e)}", None
+
 # JavaScript for export functionality
 EXPORT_JS = """
 <script>
@@ -496,9 +547,6 @@ EXPORT_JS = """
 with gr.Blocks(title="Sequential Three-Tone JoyCaption", theme=gr.themes.Soft()) as demo:
     gr.HTML(TITLE)
     
-    # JavaScript injection via HTML component
-    gr.HTML(EXPORT_JS)
-    
     with gr.Row():
         # Left column - Image and controls
         with gr.Column(scale=1):
@@ -646,7 +694,28 @@ with gr.Blocks(title="Sequential Three-Tone JoyCaption", theme=gr.themes.Soft())
             5. Generate one at a time - no simultaneous processing
             6. Copy the result and enhance it with the Text Enhancer space
             7. Use 🔄 reload to regenerate that specific tone
+            8. **📥 Export data** to use with Venice Edition Enhancer
             """)
+            
+            # Export functionality
+            with gr.Row():
+                export_btn = gr.Button(
+                    "📥 Export All Data (JSON)", 
+                    variant="primary", 
+                    size="lg"
+                )
+            
+            export_output = gr.Textbox(
+                label="Export Status",
+                lines=2,
+                interactive=False,
+                visible=False
+            )
+            
+            export_file = gr.File(
+                label="Download JSON",
+                visible=False
+            )
     
     # Individual generate button handlers
     generate_engaging_btn.click(
@@ -747,6 +816,40 @@ with gr.Blocks(title="Sequential Three-Tone JoyCaption", theme=gr.themes.Soft())
         outputs=uncensored_output
     )
     
+    # Export functionality
+    def handle_export():
+        """Handle the export button click"""
+        # Get current values from all fields
+        return export_joycaption_data(
+            keywords_input.value or "",
+            custom_instruction_input.value or "", 
+            question_input.value or "",
+            engaging_output.value or "",
+            friend_output.value or "",
+            uncensored_output.value or "",
+            qa_output.value or ""
+        )
+    
+    export_btn.click(
+        export_joycaption_data,
+        inputs=[
+            keywords_input,
+            custom_instruction_input,
+            question_input, 
+            engaging_output,
+            friend_output,
+            uncensored_output,
+            qa_output
+        ],
+        outputs=[export_output, export_file]
+    ).then(
+        lambda: gr.update(visible=True),
+        outputs=[export_output]
+    ).then(
+        lambda: gr.update(visible=True),
+        outputs=[export_file]
+    )
+    
     # Body Parts Focus clear button handler removed
 
 if __name__ == "__main__":