Update app.py

app.py

@@ -1,24 +1,23 @@
 import gradio as gr
 import os
 from groq import Groq
-import uuid
 import markdown
 
-def generate_mba_content(topic, api_key):
-    # Setting up the system prompt (assumed to be securely stored in environment variable)
-    system_prompt= os.environ["system_prompt"]
+# Retrieve system prompts from environment variables
+GENERATE_PROMPT = os.environ.get("system_prompt")
+HUMANIZE_PROMPT = os.environ.get("HUMANIZE_PROMPT")
 
-    # Initialize Groq client with user-provided API key
+def generate_mba_content(topic, api_key):
+    if not GENERATE_PROMPT:
+        return "Error: Generate prompt not found. Ensure 'GENERATE_PROMPT' is set in your environment."
+    
     try:
         client = Groq(api_key=api_key)
     except Exception as e:
         return f"Error: Failed to initialize Groq client. Check your API key. Details: {str(e)}"
 
-    # Replace [TOPIC] in the system prompt with the user-provided topic
-    prompt = system_prompt.replace("[TOPIC]", topic)
-
+    prompt = GENERATE_PROMPT.replace("[TOPIC]", topic)
     try:
-        # Generate content using Groq API
         response = client.chat.completions.create(
             model="llama3-70b-8192",
             messages=[
@@ -29,53 +28,77 @@ def generate_mba_content(topic, api_key):
             max_tokens=4000
         )
         content = response.choices[0].message.content
-        # Convert to markdown if not already in the correct format
+        content = content.replace("—", ",")  # Replace em dashes with commas
         if not content.startswith("#"):
             content = markdown.markdown(content)
         return content
     except Exception as e:
         return f"Error: Failed to generate content. Details: {str(e)}"
     finally:
-        # Clear API key from memory
         api_key = None
         if 'client' in locals():
             del client
 
-# Define Gradio interface
+def humanize_text(text, api_key):
+    if not HUMANIZE_PROMPT:
+        return "Error: Humanize prompt not found. Ensure 'HUMANIZE_PROMPT' is set in your environment."
+    
+    try:
+        client = Groq(api_key=api_key)
+    except Exception as e:
+        return f"Error: Failed to initialize Groq client. Check your API key. Details: {str(e)}"
+
+    prompt = HUMANIZE_PROMPT.replace("[TEXT]", text)
+    try:
+        response = client.chat.completions.create(
+            model="llama3-70b-8192",
+            messages=[
+                {"role": "system", "content": prompt},
+                {"role": "user", "content": f"Rewrite the following text: {text}"}
+            ],
+            temperature=0.7,
+            max_tokens=4000
+        )
+        content = response.choices[0].message.content
+        content = content.replace("—", ",")  # Replace em dashes with commas
+        if not content.startswith("#"):
+            content = markdown.markdown(content)
+        return content
+    except Exception as e:
+        return f"Error: Failed to humanize text. Details: {str(e)}"
+    finally:
+        api_key = None
+        if 'client' in locals():
+            del client
+
+# Define Gradio interface with tabs
 def gradio_app():
-    with gr.Blocks(title="MBA Content Generator") as app:
-        gr.Markdown("# MBA Content Generator")
-        gr.Markdown("Enter a topic and your Groq API key to generate MBA-style content in markdown format.")
+    with gr.Blocks(title="MBA Content Tools") as app:
+        gr.Markdown("# MBA Content Tools")
+        gr.Markdown("Generate or humanize content in MBA-style format.")
         
-        topic_input = gr.Textbox(
-            label="Topic",
-            placeholder="e.g., Strategic Management, Supply Chain Optimization",
-            lines=1
-        )
-        api_key_input = gr.Textbox(
-            label="Groq API Key",
-            placeholder="Enter your Groq API key here",
-            type="password",
-            lines=1
-        )
-        output = gr.Markdown(label="Generated Content")
-        submit_btn = gr.Button("Generate Content")
+        with gr.Tab("Generate Content from Topic"):
+            topic_input = gr.Textbox(label="Topic", placeholder="e.g., Strategic Management")
+            api_key_gen = gr.Textbox(label="Groq API Key", type="password")
+            generate_btn = gr.Button("Generate Content")
+            generate_output = gr.Markdown(label="Generated Content")
+            generate_btn.click(generate_mba_content, inputs=[topic_input, api_key_gen], outputs=generate_output)
         
-        submit_btn.click(
-            fn=generate_mba_content,
-            inputs=[topic_input, api_key_input],
-            outputs=output
-        )
+        with gr.Tab("Humanize Existing Text"):
+            text_input = gr.TextArea(label="Text to Humanize", placeholder="Paste your article, report, or paragraph here")
+            api_key_hum = gr.Textbox(label="Groq API Key", type="password")
+            humanize_btn = gr.Button("Humanize Text")
+            humanize_output = gr.Markdown(label="Humanized Content")
+            humanize_btn.click(humanize_text, inputs=[text_input, api_key_hum], outputs=humanize_output)
         
         gr.Markdown("""
         **Note:** Your API key is used securely for this session and cleared from memory afterward. 
-        This app is designed for deployment on Hugging Face Spaces. Ensure the environment variable 
-        `MBA_SYSTEM_PROMPT` is set securely in your deployment configuration.
+        Ensure the environment variables 'GENERATE_PROMPT' and 'HUMANIZE_PROMPT' are set securely in your deployment configuration.
         """)
     
     return app
 
-# Launch the app (for local testing; Hugging Face Spaces handles deployment)
+# Launch the app
 if __name__ == "__main__":
     app = gradio_app()
     app.launch()