Update app.py

app.py

@@ -36,17 +36,102 @@ def print_output(input):
     # outputs = model.generate(**inputs, max_new_tokens = 64, use_cache = True)
     # tokenizer.batch_decode(outputs)
     
-    outputs = model.generate(**inputs, max_new_tokens = 100, use_cache = True)
+    outputs = model.generate(**inputs, max_new_tokens = 200, use_cache = True, temperature=0.5)
+
+    n1 = tokenizer.batch_decode(outputs)[0].index('### Response:')
+    output_ = tokenizer.batch_decode(outputs)[0][n1+13:].replace('.com','')
+    n2 = output_.index('\n\n')
+
+    dom_names = []
+    
+    for name in output_[:n2].split('\n'):
+      if name != '':
+          dom_names.append(name.split('.')[1])
+    
+    s = '\n'.join(dom_names)
+
     return tokenizer.batch_decode(outputs)
 
 
-# Streamlit app UI
-st.title("Echo Input Example")
+# # Streamlit app UI
+# st.title("Echo Input Example")
+
+# # Text box for user input
+# user_input = st.text_input("Type something here:")
+
+# # Button to submit input
+# if st.button("Submit"):
+#     # Reprint the user input
+#     st.write(f"You typed: {print_output(user_input)}")
+
+
+def main():
+
+    css_dark_mode = """
+    <style>
+        body {
+            background-color: #121212;
+            color: #f8f9fa;
+            font-family: Arial, sans-serif;
+        }
+        .header {
+            background: linear-gradient(90deg, #005c97, #363795);
+            padding: 15px;
+            margin-bottom: 15px;
+            border-radius: 10px;
+            box-shadow: 0px 4px 8px rgba(0, 0, 0, 0.2);
+            text-align: center;
+        }
+        .header h1 {
+            color: #ffffff;
+            font-size: 36px;
+        }
+        .header h2 {
+            color: #b0b0b0;
+            font-size: 22px;
+            font-family: Georgia, serif;
+        }
+        .stButton > button {
+            background-color: #20c997;
+            color: #ffffff;
+            border: none;
+            border-radius: 8px;
+            padding: 10px 20px;
+            font-size: 16px;
+            cursor: pointer;
+            transition: background-color 0.3s;
+        }
+        .stButton > button:hover {
+            background-color: #17a589;
+        }
+        .success-box {
+            background-color: #A1D6CB;
+            padding: 10px;
+            margin: 10px 0;
+            border-radius: 8px;
+            box-shadow: 0px 4px 8px rgba(0, 0, 0, 0.2);
+        }
+    </style>
+    """
+    # Inject the custom CSS
+    st.markdown(css_dark_mode, unsafe_allow_html=True)
+
+    # Header Section
+    html_temp = """
+    <div class="header">
+        <h1>Dominious</h1>
+        <h2>AI Based Domain Name Suggestion System</h2>
+    </div>
+    """
+    st.markdown(html_temp, unsafe_allow_html=True)
+    # st.title("Domain Name Suggestion System")
 
-# Text box for user input
-user_input = st.text_input("Type something here:")
+    # Get user input
+    context = st.text_input("Describe your business here...")
 
-# Button to submit input
-if st.button("Submit"):
-    # Reprint the user input
-    st.write(f"You typed: {print_output(user_input)}")
+    # Generate the response
+    if st.button("Generate"):
+        with st.spinner("Generating Domain Names..."):
+            result = print_output(user_input)
+if __name__ == "__main__":
+    main()