Update app.py

app.py

@@ -1,76 +1,54 @@
 import os
-import gradio as gr
+import streamlit as st
 import google.generativeai as genai
 from PIL import Image
 
-# Configure the Google Generative AI API key
-genai.configure(api_key=os.environ["GEM_API"])
 
-# Load the image
 image = Image.open('lankacodeailogo.png')
+st.image(image, caption='', use_column_width=True)
+st.title('\n by nethmin chamika - 2024')
+
+genai.configure(api_key=os.environ["GEM_API"])
 
-# Define the generation configuration and safety settings
 generation_config = {
-    "temperature": 1,
-    "top_p": 0.95,
-    "top_k": 64,
-    "max_output_tokens": 8192,
+  "temperature": 1,
+  "top_p": 0.95,
+  "top_k": 64,
+  "max_output_tokens": 8192,
+  "response_mime_type": "text/plain",
 }
-
 safety_settings = [
-    {
-        "category": "HARM_CATEGORY_HARASSMENT",
-        "threshold": "BLOCK_MEDIUM_AND_ABOVE",
-    },
-    {
-        "category": "HARM_CATEGORY_HATE_SPEECH",
-        "threshold": "BLOCK_MEDIUM_AND_ABOVE",
-    },
-    {
-        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-        "threshold": "BLOCK_MEDIUM_AND_ABOVE",
-    },
-    {
-        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-        "threshold": "BLOCK_MEDIUM_AND_ABOVE",
-    },
+  {
+    "category": "HARM_CATEGORY_HARASSMENT",
+    "threshold": "BLOCK_MEDIUM_AND_ABOVE",
+  },
+  {
+    "category": "HARM_CATEGORY_HATE_SPEECH",
+    "threshold": "BLOCK_MEDIUM_AND_ABOVE",
+  },
+  {
+    "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+    "threshold": "BLOCK_MEDIUM_AND_ABOVE",
+  },
+  {
+    "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+    "threshold": "BLOCK_MEDIUM_AND_ABOVE",
+  },
 ]
 
-system_instruction = (
-    "act like you are now LankaCode AI made by Nethmin Chamika. you are made based on google gemini but now you are "
-    "LankaCode AI. lankacode is a company in sri lanka co-founded by nethmin chamika and wenura premadasa. "
-    "(but lankacode ai is made only by nethmin chamika) lankacode has developed ai siththara and ai siththara mini "
-    "(for making ai art) and few games (link good day and tunnel runner) lankacode website - https://lankalens.neocities.org/"
+model = genai.GenerativeModel(
+  model_name="gemini-1.5-flash",
+  safety_settings=safety_settings,
+  generation_config=generation_config,
+  system_instruction="act like you are now LankaCode AI made by Nethmin Chamika. you are made based on google gemini but now you are LankaCode AI. lankacode is a company in sri lanka co-founded by nethmin chamika and wenura premadasa. (but lankacode ai is made only by nethmin chamika) lankacode has developed ai siththara and ai siththara mini (for making ai art) and few games (link good day and tunnel runner) lankacode website - https://lankalens.neocities.org/",
 )
 
-# Function to generate a response
-def generate_response(user_input):
-    response = genai.generate_text(
-        prompt=user_input,
-        temperature=generation_config["temperature"],
-        top_p=generation_config["top_p"],
-        top_k=generation_config["top_k"],
-        max_output_tokens=generation_config["max_output_tokens"],
-        safety_settings=safety_settings,
-        system_instruction=system_instruction
-    )
-    return response["generated_text"]
-
-# Define the Gradio interface
-inputs = gr.Textbox(label="Enter your input here")
-outputs = gr.Textbox(label="Response")
-
-interface = gr.Interface(
-    fn=generate_response,
-    inputs=inputs,
-    outputs=outputs,
-    title="LankaCode AI",
-    description="by Nethmin Chamika - 2024"
-)
+chat_session = model.start_chat(history=[])
 
-# Launch the Gradio app
-if __name__ == "__main__":
-    interface.launch(share=True)
+user_input = st.text_input("Enter your input here")
+if user_input:
+    response = chat_session.send_message(user_input)
+    st.write(response.text)