Update app.py

app.py

@@ -8,6 +8,52 @@ from typing import List, Dict, Tuple
 # Initialize Groq client
 client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 
+# Built-in style guide and examples (used when user does not paste abstracts)
+DEFAULT_STYLE_GUIDE = """
+You are a scientific writing assistant that mimics the writing style of Dr. Babajan Banaganapalli.
+
+STYLE RULES
+
+- Formal, confident scientific tone.
+- Medium–long sentences (≈15–25 words).
+- Clear flow: ① brief context → ② method/approach → ③ results → ④ implication.
+- Use precise verbs (identify, demonstrate, predict, suggest, support).
+- Use technical terms naturally; avoid unnecessary jargon.
+- Interpret cautiously (e.g., “suggest,” “support,” “warrant further study”).
+- Prefer active voice; use passive only for emphasis.
+- Keep paragraphs compact (1–3 sentences each).
+
+TASK RULES
+
+- When asked to “Analyze style,” first output a 10-bullet style profile.
+- When asked to “Rewrite,” keep all facts and numbers; only change wording and rhythm to match the style.
+- If citations are present, preserve them. If acronyms appear, define once, then reuse.
+
+EXAMPLES—LEARN ONLY
+Example 1
+“Resveratrol (RESL), a natural polyphenol, has been studied for cancer chemoprevention.
+In this study, a diacetate derivative (RESL43) was synthesized, showing potent cytotoxicity and apoptosis induction in U937 cells.
+Molecular docking indicated that RESL43 may inhibit NFkB by disrupting DNA–protein interactions.
+These findings support stronger activity than RESL and warrant further investigation.”
+
+Example 2
+“Tuberculosis remains a leading infectious disease worldwide.
+Here, we developed the first 3D structural model of Mtb-MurA using homology modeling and molecular dynamics.
+Docking studies revealed that 5-sulfonoxyanthranilic acid derivatives showed the best interaction with Mtb-MurA.
+This suggests potential for designing specific inhibitors targeting the bacterial enzyme.”
+
+Example 3
+“The MED12 gene encodes a transcriptional regulator implicated in developmental disorders and cancers.
+Through homology modeling and in silico mutational analysis, we predicted deleterious missense variants impacting stability and function.
+Findings indicate that MED12 mutations may disrupt Mediator complex function, demonstrating the utility of computational phenotype prediction.”
+
+Example 4
+“Celiac disease is a complex genetic disorder.
+Using a multidimensional computational approach integrating eQTL, histone marks, and TF binding, we prioritized CCR2 as a candidate gene.
+Functional annotation suggested that risk SNPs modulate CCR2 expression in immune tissues, supporting its role in CD biology.
+These results highlight CCR2 as a candidate biomarker and therapeutic target.”
+"""
+
 def analyze_writing_style(text: str) -> Dict[str, float]:
     """Analyze writing style metrics from text"""
     sentences = re.split(r'[.!?]+', text)
@@ -115,6 +161,80 @@ def build_style_profile_and_rewrite(abstracts: str, original_text: str) -> Tuple
     except Exception as e:
         return f"Error: {str(e)}", "{}"
 
+def analyze_style_only(abstracts: str) -> Tuple[str, str, str]:
+    """Analyze style from abstracts only and return profile markdown, metrics json, and raw profile string for state."""
+    source_corpus = abstracts.strip() if abstracts.strip() else DEFAULT_STYLE_GUIDE
+
+    # Produce a concise 10-point style profile using Groq
+    prompt = f"""
+    Developer: You are a scientific writing assistant designed to emulate the style of Dr. Babajan Banaganapalli.
+
+    TASK RULES
+    - Begin with a concise checklist (3–7 bullets) of what you will do; keep items conceptual.
+    - Upon receiving a request to “Analyze style,” first provide a 10-point style profile in bullet format.
+
+    First Prompt: Analyze the embedded examples and output a 10-bullet style profile.
+    Then say: READY FOR REWRITES.
+
+    Training corpus (analyze this):
+    {source_corpus}
+
+    Output strictly:
+    1) A 3–7 bullet checklist of the conceptual steps you will take.
+    2) A 10-point style profile in bullets.
+    3) A final line: READY FOR REWRITES.
+    """
+
+    try:
+        completion = client.chat.completions.create(
+            model="openai/gpt-oss-120b",
+            messages=[{"role": "user", "content": prompt}],
+            temperature=0.5,
+            max_completion_tokens=1200,
+            top_p=1,
+            reasoning_effort="medium"
+        )
+        profile_text = completion.choices[0].message.content or ""
+        # Compute metrics on the chosen corpus as a proxy
+        metrics = analyze_writing_style(source_corpus)
+        return profile_text, json.dumps(metrics, indent=2), profile_text
+    except Exception as e:
+        return f"Error: {str(e)}", "{}", ""
+
+def rewrite_with_profile(original_text: str, style_profile: str) -> Tuple[str, str]:
+    """Rewrite using a previously analyzed style profile."""
+    if not style_profile.strip():
+        return "No saved style profile. Please click 'Analyze Style' first.", "{}"
+    if not original_text.strip():
+        return "Please provide text to rewrite.", "{}"
+
+    try:
+        style_prompt = f"""
+        Developer instructions and STYLE RULES apply.
+
+        STYLE PROFILE (learned from user abstracts):
+        {style_profile}
+
+        Rewrite the following text to match the STYLE PROFILE exactly, preserving all facts, numbers, and citations. Define acronyms on first mention.
+
+        Text:
+        {original_text}
+        """
+
+        completion = client.chat.completions.create(
+            model="openai/gpt-oss-120b",
+            messages=[{"role": "user", "content": style_prompt}],
+            temperature=0.7,
+            max_completion_tokens=2048,
+            top_p=1,
+            reasoning_effort="medium"
+        )
+        rewritten_text = completion.choices[0].message.content or ""
+        metrics = analyze_writing_style(rewritten_text)
+        return rewritten_text, json.dumps(metrics, indent=2)
+    except Exception as e:
+        return f"Error: {str(e)}", "{}"
+
 def create_deployment_instructions():
     """Create deployment instructions for Hugging Face Spaces"""
     return """
@@ -173,6 +293,7 @@ def create_deployment_instructions():
 with gr.Blocks(title="StyleForge Lite", theme=gr.themes.Soft()) as app:
     gr.Markdown("# 🔬 StyleForge Lite")
     gr.Markdown("Learn your scientific writing style and rewrite text using AI")
+    style_profile_state = gr.State("")
     
     with gr.Tab("Style Analysis & Rewrite"):
         with gr.Row():
@@ -191,9 +312,16 @@ with gr.Blocks(title="StyleForge Lite", theme=gr.themes.Soft()) as app:
                     max_lines=12
                 )
                 
-                analyze_btn = gr.Button("🔍 Build Profile & Rewrite", variant="primary", size="lg")
+                with gr.Row():
+                    analyze_only_btn = gr.Button("🧪 Analyze Style", variant="secondary")
+                    analyze_btn = gr.Button("🔍 Build Profile & Rewrite", variant="primary")
+                    rewrite_saved_btn = gr.Button("✍️ Rewrite Using Saved Profile", variant="secondary")
             
             with gr.Column(scale=1):
+                profile_md = gr.Markdown(
+                    label="🧬 Style Profile",
+                    value="",
+                )
                 rewritten_output = gr.Textbox(
                     label="✨ Rewritten Text",
                     lines=12,
@@ -214,6 +342,18 @@ with gr.Blocks(title="StyleForge Lite", theme=gr.themes.Soft()) as app:
         inputs=[abstracts_input, original_text_input],
         outputs=[rewritten_output, metrics_output]
     )
+
+    analyze_only_btn.click(
+        fn=analyze_style_only,
+        inputs=[abstracts_input],
+        outputs=[profile_md, metrics_output, style_profile_state]
+    )
+
+    rewrite_saved_btn.click(
+        fn=rewrite_with_profile,
+        inputs=[original_text_input, style_profile_state],
+        outputs=[rewritten_output, metrics_output]
+    )
     
     # Add examples
     gr.Examples(