Create app.py

app.py

@@ -0,0 +1,109 @@
+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.getenv("MBA_SYSTEM_PROMPT", """
+Generate content in markdown format for an MBA student on the topic of [TOPIC]. The content should feel human-written, as if crafted by an MBA student for a submission, presentation, or casual peer discussion. Follow these rules:
+
+Persona: Write as an MBA student sharing insights with classmates. Keep the tone casual and friendly, like a conversation over pizza after class. Add a touch of humor or a personal story where it fits.
+
+Constraints:
+- No em dashes. Use commas, parentheses, or colons instead.
+- Skip stiff phrases like "It is imperative" or "This paper will explore". Go for relaxed options like "Here’s the deal" or "Let’s dig into this".
+- Limit transition words (e.g., "moreover", "thus").
+- Avoid these terms: aligns, aims to explore, today’s fast-paced world, notable works include, notable figures, surpassing, tragically, impacting, making an impact, research needed to understand, despite facing, expressed excitement, evolving situation, delve, harnessing, at the heart of, in essence, facilitating, intrinsic, integral, core, facet, nuance, culmination, manifestation, inherent, confluence, underlying, intricacies, epitomize, embodiment, iteration, synthesize, amplify, impetus, catalyst, synergy, cohesive, paradigm, dynamics, implications, prerequisite, fusion, holistic, quintessential, cohesion, symbiosis, integration, encompass, unveil, unravel, emanate, illuminate, reverberate, augment, infuse, extrapolate, embody, unify, inflection, instigate, embark, envisage, elucidate, substantiate, resonate, catalyze, resilience, evoke, pinnacle, evolve, digital bazaar, tapestry, leverage, centerpiece, subtlety, immanent, exemplify, blend, comprehensive, archetypal, unity, harmony, conceptualize, reinforce, mosaic, catering.
+
+Style:
+- Mix up sentence lengths: some short and snappy, others longer with detail.
+- Use contractions (e.g., can’t, you’re) and casual phrases that sound natural.
+- Stick to active voice (e.g., "I learned" not "It was learned").
+- Add sensory details (e.g., "The coffee burned my tongue while I scrambled to finish the slide deck").
+- Include a personal anecdote, hypothetical situation, or relatable example (e.g., a chaotic group project or a real-world internship moment).
+
+Structure:
+- Introduction: Kick off with a question, a fun fact, or a relatable moment (e.g., pulling an all-nighter before a deadline).
+- Body: Use markdown subheadings (e.g., ## My Take) if needed. Include 1-2 anecdotes, scenarios, or examples to back up your points.
+- Conclusion: Wrap up with practical advice or a reflective thought that sticks with the reader.
+
+Examples:
+Unacceptable: "In today’s fast-paced world, leveraging technology is key, furthermore, it facilitates synergy across teams."
+Acceptable: "Tech’s a game-changer, no doubt. Last semester, my team used Slack to avoid those endless email chains, and we actually got stuff done without losing our minds."
+
+Output the content in markdown format.
+""")
+
+    # Initialize Groq client with user-provided API key
+    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)
+
+    try:
+        # Generate content using Groq API
+        response = client.chat.completions.create(
+            model="llama3-70b-8192",
+            messages=[
+                {"role": "system", "content": prompt},
+                {"role": "user", "content": f"Generate content for the topic: {topic}"}
+            ],
+            temperature=0.7,
+            max_tokens=1000
+        )
+        content = response.choices[0].message.content
+        # Convert to markdown if not already in the correct format
+        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 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.")
+        
+        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")
+        
+        submit_btn.click(
+            fn=generate_mba_content,
+            inputs=[topic_input, api_key_input],
+            outputs=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.
+        """)
+    
+    return app
+
+# Launch the app (for local testing; Hugging Face Spaces handles deployment)
+if __name__ == "__main__":
+    app = gradio_app()
+    app.launch()