Update app.py

app.py

@@ -2,7 +2,7 @@ import sys
 import os
 import types
 
-# ── Python 3.13 fix: stub out audioop ───────────────────────────────────────
+# Python 3.13 audioop stub (not needed on 3.11 but harmless)
 if 'audioop' not in sys.modules:
     sys.modules['audioop'] = types.ModuleType('audioop')
 
@@ -12,19 +12,15 @@ import matplotlib.pyplot as plt
 import matplotlib
 matplotlib.use('Agg')
 
-# ── Model loading ────────────────────────────────────────────────────────────
 model = None
 
 def load_model():
     global model
     if model is not None:
-        return "✅ Model already loaded!"
+        return "✅ Already loaded!"
     try:
         from tribev2 import TribeModel
-        model = TribeModel.from_pretrained(
-            "facebook/tribev2",
-            cache_folder="./tribe_cache"
-        )
+        model = TribeModel.from_pretrained("facebook/tribev2", cache_folder="./tribe_cache")
         return "✅ Model loaded!"
     except Exception as e:
         return f"❌ Error: {str(e)}"
@@ -46,8 +42,7 @@ def score_predictions(preds):
     for name, s, e, _ in REGIONS:
         start, end = int(half * s), int(half * e)
         scores[name] = round(float(np.mean(avg[start:end]) / global_max * 100), 1)
-    overall = round(sum(scores.values()) / len(scores), 1)
-    return scores, overall
+    return scores, round(sum(scores.values()) / len(scores), 1)
 
 def make_brain_plot(preds):
     try:
@@ -63,10 +58,9 @@ def make_brain_plot(preds):
         plotting.plot_surf_stat_map(fsaverage.infl_right, avg_norm[half:], hemi="right",
             view="lateral", colorbar=True, cmap="hot", title="Right hemisphere", axes=axes[1], figure=fig)
         plt.tight_layout()
-        path = "./brain_map.png"
-        plt.savefig(path, dpi=130, bbox_inches="tight", facecolor="#111111")
+        plt.savefig("/tmp/brain_map.png", dpi=130, bbox_inches="tight", facecolor="#111111")
         plt.close()
-        return path
+        return "/tmp/brain_map.png"
     except Exception as e:
         print(f"Brain plot error: {e}")
         return None
@@ -91,10 +85,9 @@ def make_score_chart(scores, overall):
         ax.text(bar.get_width() + 1, bar.get_y() + bar.get_height() / 2,
                 f"{val}", va="center", color="white", fontsize=10, fontweight="bold")
     plt.tight_layout()
-    path = "./score_chart.png"
-    plt.savefig(path, dpi=130, bbox_inches="tight", facecolor="#1a1a1a")
+    plt.savefig("/tmp/score_chart.png", dpi=130, bbox_inches="tight", facecolor="#1a1a1a")
     plt.close()
-    return path
+    return "/tmp/score_chart.png"
 
 def generate_suggestions(scores, overall):
     tips = []
@@ -127,20 +120,19 @@ def analyze_script(script_text, progress=gr.Progress()):
         from gtts import gTTS
         progress(0.2, desc="Converting script to speech...")
         tts = gTTS(text=script_text.strip(), lang="en", slow=False)
-        audio_path = "./script_audio.mp3"
-        tts.save(audio_path)
+        tts.save("/tmp/script_audio.mp3")
         progress(0.4, desc="Running TRIBE v2 prediction (1-3 mins)...")
-        df = model.get_events_dataframe(audio_path=audio_path)
+        df = model.get_events_dataframe(audio_path="/tmp/script_audio.mp3")
         preds, segments = model.predict(events=df)
         progress(0.7, desc="Scoring regions...")
         scores, overall = score_predictions(preds)
         progress(0.8, desc="Rendering maps...")
-        brain_img  = make_brain_plot(preds)
-        score_img  = make_score_chart(scores, overall)
+        brain_img   = make_brain_plot(preds)
+        score_img   = make_score_chart(scores, overall)
         suggestions = generate_suggestions(scores, overall)
-        np.save("./brain_predictions.npy", preds)
+        np.save("/tmp/brain_predictions.npy", preds)
         progress(1.0, desc="Done!")
-        return brain_img, score_img, suggestions, "./brain_predictions.npy"
+        return brain_img, score_img, suggestions, "/tmp/brain_predictions.npy"
     except Exception as e:
         return None, None, f"❌ Error:\n{str(e)}", None
 
@@ -148,14 +140,11 @@ css = "#title{text-align:center} #subtitle{text-align:center;color:#888;font-siz
 
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="indigo"), css=css) as demo:
     gr.Markdown("# 🧠 Script Brain Optimizer", elem_id="title")
-    gr.Markdown("Paste your script → get real fMRI predictions via **TRIBE v2** → iterate", elem_id="subtitle")
+    gr.Markdown("Paste your script → real fMRI predictions via **TRIBE v2** → iterate", elem_id="subtitle")
     with gr.Row():
         with gr.Column(scale=1):
-            script_input = gr.Textbox(
-                label="Your script",
-                placeholder="Paste your content script here...",
-                lines=12, max_lines=20
-            )
+            script_input = gr.Textbox(label="Your script",
+                placeholder="Paste your content script here...", lines=12, max_lines=20)
             with gr.Row():
                 clear_btn   = gr.Button("Clear", variant="secondary", scale=1)
                 analyze_btn = gr.Button("🧠 Analyze", variant="primary", scale=3)
@@ -165,16 +154,11 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="indigo"), css=css) as demo:
             brain_img_out = gr.Image(label="Brain activation map", height=320)
             score_img_out = gr.Image(label="Region scores", height=280)
 
-    analyze_btn.click(
-        fn=analyze_script,
-        inputs=[script_input],
-        outputs=[brain_img_out, score_img_out, suggestions_out, download_out]
-    )
-    clear_btn.click(
-        fn=lambda: ("", None, None, "*Paste a script and click Analyze...*", None),
-        outputs=[script_input, brain_img_out, score_img_out, suggestions_out, download_out]
-    )
+    analyze_btn.click(fn=analyze_script, inputs=[script_input],
+                      outputs=[brain_img_out, score_img_out, suggestions_out, download_out])
+    clear_btn.click(fn=lambda: ("", None, None, "*Paste a script and click Analyze...*", None),
+                    outputs=[script_input, brain_img_out, score_img_out, suggestions_out, download_out])
     gr.Markdown("---\n*Powered by [TRIBE v2](https://github.com/facebookresearch/tribev2) by Meta FAIR*")
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(server_name="0.0.0.0", server_port=7860)