Update app.py

app.py

@@ -6,15 +6,17 @@ from PIL import Image
 # Configure the Google Generative AI API key
 genai.configure(api_key=os.environ["GEM_API"])
 
-# Define the generation configuration
+# Load the image
+image = Image.open('lankacodeailogo.png')
+
+# Define the generation configuration and safety settings
 generation_config = {
     "temperature": 1,
     "top_p": 0.95,
     "top_k": 64,
     "max_output_tokens": 8192,
-    "response_mime_type": "text/plain",
 }
-# Define the safety settings
+
 safety_settings = [
     {
         "category": "HARM_CATEGORY_HARASSMENT",
@@ -34,41 +36,39 @@ safety_settings = [
     },
 ]
 
-# Initialize the generative model
-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/",
+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/"
 )
 
-chat_session = model.start_chat(history=[])
-
-# Define the function to generate a response
+# Function to generate a response
 def generate_response(user_input):
-    response = chat_session.send_message(user_input)
-    return response.text
+    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
 
-# Load the image
-image = Image.open('lankacodeailogo.png')
-
-# Create a Gradio interface
+# Gradio interface
 interface = gr.Interface(
     fn=generate_response,
     inputs=gr.inputs.Textbox(label="Enter your input here"),
     outputs=gr.outputs.Textbox(label="Response"),
     title="LankaCode AI",
     description="by Nethmin Chamika - 2024",
-    theme="compact",
-    server_name="0.0.0.0",
-    server_port=7860
+    theme="compact"
 )
 
-# Display the image on the Gradio interface
-def display_image():
-    return image
-
 # Launch the Gradio app
 if __name__ == "__main__":
     interface.launch(share=True)
 
+
+