Update app.py

app.py

@@ -1,106 +0,0 @@
-# Hey its your saiyan Utkarsh Shukla. I gonna write my custom comments after each line of code. So even a beginner (previous me) can read and get whats going on.
-# Bear with me, I got your Back buddy
-
-import streamlit as st  
-# The saviour web app creator, easy peasy web app creation by few lines of codes. 
-# No HTML, CSS, or JS needed!"
-
-from transformers import AutoTokenizer, AutoModelForCausalLM
-# transformers here is just library which gives us access to Transformer architecture based pretrained models for natural language processing and other tasks. For now, Think of Tranformer Architecture, as mystery Architecture which makes models way cooler.
-
-
-# AutoTokenizer helps in Text input -> Sentences -> Words -> Even subwords like ['un', 'break', 'able'] -> Integer IDs that model expects.
-# And whats awesome is Tokens will be generated following the configurations and requirements of model which we will be using.
-
-# AutoModelForCausalLM is a powerful and convenient class serves as a high-level interface for loading pre-trained transformer models specifically designed for causal language modeling.
-# The cooler part of these Auto* classes are you don't need to know exact class name of model(like GPT2LMHeadModel, CTRLLMHeadModel, ReformerLMHeadModel, etc.).
-# Meaning The AutoModelForCausalLM automatically determine the correct model architecture based on the pretrained_model_name_or_path you provide. AutoModelForCausalLM infers this from the configuration files associated with the pre-trained model.
-
-from wordcloud import WordCloud
-# This will help us in knowing which words have large frequency. It creates a visual representation of words used, know as Word Cloud.
-# More the frequency + More the importance -> Word will appear larger in Word Cloud.
-# Mostly it avoids our stop words like it, is, are etc
-
-
-import matplotlib.pyplot as plt
-# This guy helps us to plot. So wait till you see it.
-
-import torch
-
-# Cache the model and tokenizer to avoid reloading on every run
-@st.cache_resource
-def load_model_and_tokenizer():
-    model_name = "google/gemma-2b"  # Replace with "gemma-7b" if desired
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForCausalLM.from_pretrained(model_name)
-    return tokenizer, model
-
-# Function to generate text with Gemma
-def generate_text(prompt, tone, max_length):
-    tokenizer, model = load_model_and_tokenizer()
-    # Adjust prompt based on tone
-    tone_prompts = {
-        "Funny": f"Generate a funny response to: {prompt}",
-        "Serious": f"Provide a serious and thoughtful response to: {prompt}",
-        "Poetic": f"Write a poetic response to: {prompt}"
-    }
-    input_text = tone_prompts.get(tone, prompt)
-    
-    inputs = tokenizer(input_text, return_tensors="pt")
-    outputs = model.generate(
-        inputs["input_ids"],
-        max_length=max_length + len(input_text.split()),  # Account for prompt length
-        num_return_sequences=1,
-        temperature=0.7,  # Creativity level
-        do_sample=True
-    )
-    return tokenizer.decode(outputs[0], skip_special_tokens=True)
-
-# Custom CSS for styling
-st.markdown("""
-    <style>
-    body {background-color: #f0f0f0;}
-    .title {color: #2c3e50; font-size: 36px; font-weight: bold;}
-    .instructions {color: #34495e; font-size: 18px;}
-    .output-box {background-color: #ecf0f1; padding: 10px; border-radius: 5px;}
-    </style>
-""", unsafe_allow_html=True)
-
-# App header with image
-st.image("https://unsplash.com/photos/8xznAGy4HcY/download?force=true&w=640", caption="AI in Action")
-st.markdown('<p class="title">Gemma Text Generator</p>', unsafe_allow_html=True)
-
-# Instructions and example
-st.markdown("""
-    <p class="instructions">
-    Enter a prompt below to generate text using the Gemma model from DeepMind. Customize the tone and length to see different outputs!<br>
-    <b>Example:</b> Prompt: "The cat sat on" | Tone: "Funny" | Length: 50 → "The cat sat on my homework and laughed as I cried over my grades."
-    </p>
-""", unsafe_allow_html=True)
-
-# User input section
-with st.form(key="input_form"):
-    prompt = st.text_input("Enter a prompt", placeholder="e.g., 'The future of AI is'")
-    tone = st.selectbox("Tone", ["Funny", "Serious", "Poetic"])
-    max_length = st.slider("Word count", 20, 100, 50)
-    submit_button = st.form_submit_button(label="Generate")
-
-# Generate and display output
-if submit_button:
-    if not prompt:
-        st.error("Please enter a prompt!")
-    else:
-        with st.spinner("Generating text..."):
-            output = generate_text(prompt, tone, max_length)
-            st.markdown(f'<div class="output-box">{output}</div>', unsafe_allow_html=True)
-            
-            # Generate and display word cloud
-            wordcloud = WordCloud(width=400, height=200, background_color="white").generate(output)
-            plt.figure(figsize=(8, 4))
-            plt.imshow(wordcloud, interpolation="bilinear")
-            plt.axis("off")
-            st.pyplot(plt)
-
-# Footer
-st.markdown("---")
-st.write("Built with ❤️ by Saiyan Utkarsh Shukla for GSoC 2025 | Powered by Gemma and Hugging Face")