DNA-Diffusion

Runtime error

App Files Files Community

openfree commited on Jul 2

Commit

da3b7ce

verified ·

1 Parent(s): ea36a1d

Update app.py

Browse files

Files changed (1) hide show

app.py +81 -12

app.py CHANGED Viewed

@@ -12,6 +12,7 @@ import time
 import random
 import numpy as np
 from datetime import datetime
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
@@ -176,6 +177,19 @@ class DNADiffusionApp:
 app = DNADiffusionApp()
 # HTML for 3D Viewer
 HTML_3D_VIEWER = """
 <!DOCTYPE html>
@@ -466,25 +480,41 @@ def create_demo():
     css = """
     .gradio-container {
         font-family: 'Arial', sans-serif;
     }
     .sequence-box {
         font-family: 'Courier New', monospace;
-        background-color: #f0f0f0;
-        padding: 10px;
-        border-radius: 5px;
     }
     iframe {
         border: none;
         border-radius: 10px;
     }
     """
-    with gr.Blocks(css=css, title="DNA-Diffusion with 3D Viewer") as demo:
         gr.Markdown(
             """
-            # 🧬 DNA-Diffusion: AI-Powered Sequence Generation with 3D Visualization
-            Generate cell-type specific DNA sequences and visualize them in 3D!
             """
         )
@@ -545,8 +575,9 @@ def create_demo():
                         gr.Markdown("The 3D viewer shows the double helix structure of your generated DNA sequence.")
                         # HTML component for 3D viewer
                         viewer_html = gr.HTML(
-                            value=f'<iframe src="data:text/html;charset=utf-8,{HTML_3D_VIEWER.replace("#", "%23")}" width="100%" height="600px"></iframe>',
                             label="3D Molecular Viewer"
                         )
@@ -556,7 +587,27 @@ def create_demo():
                             variant="secondary"
                         )
-            # Tab 3: Batch Generation
             with gr.TabItem("📦 Batch Generation"):
                 with gr.Row():
                     with gr.Column():
@@ -609,7 +660,7 @@ def create_demo():
         def update_3d_viewer(sequence):
             if not sequence:
-                return gr.HTML.update()
             # Create HTML with embedded sequence data
             html_with_sequence = HTML_3D_VIEWER.replace(
@@ -617,9 +668,9 @@ def create_demo():
                 f"updateSequence('{sequence}');\n        window.addEventListener('message'"
             )
-            return gr.HTML.update(
-                value=f'<iframe src="data:text/html;charset=utf-8,{html_with_sequence.replace("#", "%23")}" width="100%" height="600px"></iframe>'
-            )
         def generate_batch(cell_types, count):
             if not cell_types:
@@ -675,6 +726,24 @@ def create_demo():
             inputs=[batch_cell_types, batch_count],
             outputs=[batch_output, status_text]
         )
     return demo

 import random
 import numpy as np
 from datetime import datetime
+import urllib.parse
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
 app = DNADiffusionApp()
+def load_html_file(filename):
+    """Safely load and prepare HTML file for iframe embedding"""
+    try:
+        with open(filename, 'r', encoding='utf-8') as f:
+            content = f.read()
+        # Properly encode content for data URI
+        encoded_content = urllib.parse.quote(content, safe='')
+        return f'<iframe src="data:text/html;charset=utf-8,{encoded_content}" width="100%" height="800px" style="border: none; border-radius: 10px;"></iframe>'
+    except FileNotFoundError:
+        return f'<div style="padding: 40px; text-align: center; color: #ff6b6b; background: #2a2a2a; border-radius: 10px;"><h3>⚠️ File Not Found</h3><p>{filename} not found in the current directory.</p></div>'
+    except Exception as e:
+        return f'<div style="padding: 40px; text-align: center; color: #ff6b6b; background: #2a2a2a; border-radius: 10px;"><h3>❌ Error Loading File</h3><p>{str(e)}</p></div>'
 # HTML for 3D Viewer
 HTML_3D_VIEWER = """
 <!DOCTYPE html>
     css = """
     .gradio-container {
         font-family: 'Arial', sans-serif;
+        background: linear-gradient(135deg, #0a0a0a 0%, #1a1a1a 100%);
     }
     .sequence-box {
         font-family: 'Courier New', monospace;
+        background-color: #1e1e1e;
+        color: #00ff88;
+        padding: 15px;
+        border-radius: 8px;
+        border: 1px solid #00ff88;
     }
     iframe {
         border: none;
         border-radius: 10px;
+        box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+    }
+    .gr-button-primary {
+        background: linear-gradient(135deg, #00ff88, #0088ff) !important;
+        border: none !important;
+        color: #000 !important;
+    }
+    .gr-button-secondary {
+        background: linear-gradient(135deg, #ff6b6b, #ff9f40) !important;
+        border: none !important;
+        color: #fff !important;
     }
     """
+    with gr.Blocks(css=css, title="DNA-Diffusion Suite", theme=gr.themes.Soft()) as demo:
         gr.Markdown(
             """
+            # 🧬 DNA-Diffusion Suite: AI-Powered Molecular Engineering
+            Generate, visualize, and gamify DNA sequences with advanced AI models!
+            [![GPU Status](https://img.shields.io/badge/GPU-Enabled-brightgreen)](https://huggingface.co/spaces)
             """
         )
                         gr.Markdown("The 3D viewer shows the double helix structure of your generated DNA sequence.")
                         # HTML component for 3D viewer
+                        encoded_viewer = urllib.parse.quote(HTML_3D_VIEWER, safe='')
                         viewer_html = gr.HTML(
+                            value=f'<iframe src="data:text/html;charset=utf-8,{encoded_viewer}" width="100%" height="600px"></iframe>',
                             label="3D Molecular Viewer"
                         )
                             variant="secondary"
                         )
+            # Tab 3: DNA Casino (if external HTML exists)
+            with gr.TabItem("🎰 DNA Casino"):
+                gr.Markdown("### DNA Slot Machine Game")
+                gr.Markdown("Experience DNA generation as a casino game!")
+                # Load external HTML file using helper function
+                casino_html = gr.HTML(
+                    value=load_html_file('dna-slot-machine.html'),
+                    label="DNA Casino Game"
+                )
+                # Button to capture sequence from casino
+                with gr.Row():
+                    casino_sequence = gr.Textbox(
+                        label="Casino Generated Sequence",
+                        placeholder="Sequence from casino will appear here",
+                        interactive=False
+                    )
+                    analyze_casino_btn = gr.Button("🔬 Analyze Casino Sequence", variant="secondary")
+            # Tab 4: Batch Generation
             with gr.TabItem("📦 Batch Generation"):
                 with gr.Row():
                     with gr.Column():
         def update_3d_viewer(sequence):
             if not sequence:
+                return viewer_html.value
             # Create HTML with embedded sequence data
             html_with_sequence = HTML_3D_VIEWER.replace(
                 f"updateSequence('{sequence}');\n        window.addEventListener('message'"
             )
+            # Properly encode content for data URI
+            encoded_content = urllib.parse.quote(html_with_sequence, safe='')
+            return f'<iframe src="data:text/html;charset=utf-8,{encoded_content}" width="100%" height="600px"></iframe>'
         def generate_batch(cell_types, count):
             if not cell_types:
             inputs=[batch_cell_types, batch_count],
             outputs=[batch_output, status_text]
         )
+        # Casino sequence analysis
+        def analyze_casino_sequence(seq):
+            if not seq:
+                return {}, "❌ No sequence to analyze"
+            analysis = app.analyze_sequence(seq)
+            return analysis, "✅ Casino sequence analyzed"
+        try:
+            analyze_casino_btn.click(
+                fn=analyze_casino_sequence,
+                inputs=[casino_sequence],
+                outputs=[analysis_output, status_text]
+            )
+        except NameError:
+            # Casino tab components may not be defined if HTML file is missing
+            pass
     return demo