Update app.py

app.py

@@ -1,61 +1,54 @@
+from dotenv import load_dotenv
+load_dotenv()
 import streamlit as st
+import os
+import google.generativeai as genai
+from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain.prompts import PromptTemplate
-from langchain.llms import CTransformers
-# from transformers import AutoModel
-
-## Function To get response from LLAma 2 model
-
-def getLLamaresponse(input_text,no_words,blog_style):
-
-    # Load model directly
-
-    ### LLama2 model
-    llm=CTransformers(model='models/llama-2-7b-chat.ggmlv3.q8_0.bin',
-                      model_type='llama',
-                      config={'max_new_tokens':256,
-                              'temperature':0.01})
-    
-    ## Prompt Template
-
-    template="""
-        Write a blog for {blog_style} profile for a topic {input_text}
-        within {no_words} words.
-            """
-    
-    prompt=PromptTemplate(input_variables=["blog_style","input_text",'no_words'],
-                          template=template)
-    
-    ## Generate the ressponse from the LLama 2 model
-    response=llm(prompt.format(blog_style=blog_style,input_text=input_text,no_words=no_words))
-    print(response)
-    return response
-
 
+genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
+model = ChatGoogleGenerativeAI(model="gemini-pro")
 
+def gemini_model(input_text,no_of_words,blog_style):
+    # here we are creating a template for the prompt
+    template = """
+    Write a blog on the topic of {input_text} for {blog_style} audience. 
+    The blog should be {no_of_words} words long.
+    """
 
+    # here we are creating a prompt using the template and the input variables
+    prompt = PromptTemplate(input_variables=["input_text","blog_style","no_of_words"],template=template)
 
+    # here we are generating the blog
+    response = model.invoke(prompt.format(input_text=input_text,blog_style=blog_style,no_of_words=no_of_words))
+    print(response)
+    return response.content
 
-st.set_page_config(page_title="Generate Blogs",
-                    page_icon='♜',
-                    layout='centered',
-                    initial_sidebar_state='collapsed')
 
-st.header("Lets Generate Blogs Together ♜")
+st.set_page_config(page_title="Blog Generator", initial_sidebar_state="collapsed", layout="centered")
 
-input_text=st.text_input("Enter the Blog Topic")
+# Header
+st.title("📝 Generate Blog")
 
-## creating to more columns for additonal 2 fields
+# Input Section
+input_text = st.text_input("🔍 Enter the topic of the blog you want to generate")
 
-col1,col2=st.columns([5,5])
+# Creating 2 columns for additional 2 fields
+col1, col2 = st.columns([2, 2])
 
+# Number of words input
 with col1:
-    no_words=st.text_input('No of Words')
+    no_of_words = st.text_input("📏 Number of Words", value="500")
+
+# Blog style selection
 with col2:
-    blog_style=st.selectbox('Writing the blog for',
-                            ('Kids - 5 Year olds','Grad Students','Researchers','Data Scientist','Common People',),index=0)
-    
-submit=st.button("Generate")
-
-## Final response
-if submit:
-    st.write(getLLamaresponse(input_text,no_words,blog_style))
+    blog_style = st.selectbox("📝 Writing the blog for", ("Researchers or Professionals", "General Audience"), index=0)
+
+# Generate Button
+submit_button = st.button("Generate Blog 🚀")
+
+# Display the generated blog on button click
+if submit_button:
+    st.success("📖 **Generated Blog:**")
+    generated_blog = gemini_model(input_text, no_of_words, blog_style)
+    st.write(generated_blog)