submission-team-primus

README.md

@@ -1,26 +1,71 @@
-# Gen-Ai-Rush-Buildathon
-
-## Submission Instruction:
-  1. Fork this repository
-  2. Create a folder with your Team Name
-  3. Upload all the code and necessary files in the created folder
-  4. Upload a **README.md** file in your folder with the below mentioned informations.
-  5. Generate a Pull Request with your Team Name. (Example: submission-XYZ_team)
-
-## README.md must consist of the following information:
-
-#### Team Name -
-#### Problem Statement - 
-#### Team Leader Email -
-
-## A Brief of the Prototype:
-  This section must include UML Diagrams and prototype description
-
-## Tech Stack: 
-   List Down all technologies used to Build the prototype
-
-## Step-by-Step Code Execution Instructions:
-  This Section must contain a set of instructions required to clone and run the prototype so that it can be tested and deeply analyzed
-
-## What I Learned:
-   Write about the biggest learning you had while developing the prototype
+# 3D-VizAi - Visualize Your Imagination
+
+![3D-VizAi Logo](logo.png)
+
+## Introduction
+
+3D-VizAi is an innovative platform that combines the power of Artificial Intelligence and creativity to bring your ideas to life. This repository contains the source code for 3D-VizAi, where you can explore various features like Text-to-2D, Text-to-3D, and Image-to-Text, enabling you to visualize your imagination in captivating ways.
+
+## Getting Started
+
+To begin your journey with 3D-VizAi, follow these simple steps:
+
+1. Clone this repository to your local machine using the following command:
+
+Sure! Here's the content for the README file based on the video description:
+
+markdown
+Copy code
+# 3D-VizAi - Visualize Your Imagination
+
+![3D-VizAi Logo](logo.png)
+
+## Introduction
+
+3D-VizAi is an innovative platform that combines the power of Artificial Intelligence and creativity to bring your ideas to life. This repository contains the source code for 3D-VizAi, where you can explore various features like Text-to-2D, Text-to-3D, and Image-to-Text, enabling you to visualize your imagination in captivating ways.
+
+## Getting Started
+
+To begin your journey with 3D-VizAi, follow these simple steps:
+
+
+Install the required dependencies by navigating to the project directory and running:
+Register or login to your 3D-VizAi account to unleash the magic of creativity!
+
+## Features
+
+### Text-to-2D
+
+Express your ideas in words and watch them transform into stunning 2D images. With our AI-powered image synthesis algorithms, your text prompts come to life with a touch of enchantment. Simply type your text, hit submit, and discover a world of captivating images that you can download and share.
+
+### Text-to-3D
+
+Elevate your imagination to a whole new dimension with Text-to-3D. Enter your text prompts, and our advanced algorithms will craft intricate 3D models that reflect your creativity. Preview your creations from every angle and download them for an immersive experience like no other.
+
+### Image-to-Text
+
+Capture breathtaking images and let 3D-VizAi do the rest. Upload your photos, and our AI model will generate engaging one-line descriptions, enhancing your storytelling abilities. Bring depth to your images with textual narratives that leave a lasting impression.
+
+## Future Implementations
+
+We are constantly innovating to enhance your creative journey. In the near future, we have two exciting features in the pipeline:
+
+### 2D-to-3D
+
+Soon, you'll be able to take any 2D creation and elevate it to a mesmerizing 3D masterpiece. Unleash the potential of 2D-to-3D transformation and see your flat designs come alive in the third dimension.
+
+### AR View
+
+Immerse yourself in the realm of augmented reality. Step into a world where your imagination blends seamlessly with reality. AR View is on the horizon, and you won't want to miss it!
+
+## Contribution and Feedback
+
+We value the open-source community and welcome your contributions and feedback. If you encounter any issues, have ideas for improvements, or wish to contribute to the project, feel free to open an issue or submit a pull request. Together, let's shape a world of boundless imagination!
+
+## License
+
+You are free to use, modify, and distribute this project in accordance with the terms of the license.
+
+---
+
+With 3D-VizAi, the only limit is your creativity. Unleash the power of AI and embark on a journey where your imagination knows no bounds. Join us today and visualize your dreams in ways you never thought possible!

main.py

@@ -0,0 +1,206 @@
+from flask import Flask,render_template,request,redirect,url_for
+import random
+from email.mime.multipart import MIMEMultipart
+from email.mime.image import MIMEImage
+from email.mime.text import MIMEText
+import smtplib,ssl
+import numpy as np
+import mysql.connector
+import joblib
+import os
+from threedgen import model3d_generator
+
+app=Flask(__name__)
+
+mydb = mysql.connector.connect(
+  host="localhost",
+  user="root",
+  password="",
+  database="genai"
+)
+mycursor = mydb.cursor()
+#function used to send confirmation mail to the user
+def mail_send(otp,mail):
+    try:
+        s = smtplib.SMTP('smtp.office365.com', 587)
+    except Exception as e:
+        s = smtplib.SMTP_SSL('smtp.office365.com', 465)
+    s.ehlo()
+    s.starttls()
+    s.login("harikrishna.20it@sonatech.ac.in", "Rhk$14082002")
+
+    msg = MIMEMultipart()
+    msg['From']='harikrishna.20it@sonatech.ac.in'
+    msg['To']=mail
+    msg['Subject']="Registration Confirmation"
+
+    html=f'''\
+        <!DOCTYPE html>
+<html>
+<head>
+    <title></title>
+    <meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
+    <meta name="viewport" content="width=device-width, initial-scale=1">
+    <meta http-equiv="X-UA-Compatible" content="IE=edge" />
+    <style type="text/css">
+        body, table, td, a { "-webkit-text-size-adjust: 100%; -ms-text-size-adjust: 100%;" }
+
+        table, td {" mso-table-lspace: 0pt; mso-table-rspace: 0pt;" }
+
+        img { "-ms-interpolation-mode: bicubic;"  }
+        img { " border: 0; height: auto; line-height: 100%;  outline: none; text-decoration: none; " }
+        table { " border-collapse: collapse !important; " }
+        body { " height: 100% !important; margin: 0 !important; padding: 0 !important; background: #e6f4ff; " }
+        a[x-apple-data-detectors] { " color: inherit !important; text-decoration: none !important; font-size: inherit !important; font-family: inherit !important; font-weight: inherit !important; line-height: inherit !important; " }
+        div[style*="margin: 16px 0;"] { "margin: 0 !important; " }
+        .page-center { " display: flex; justify-content: center; align-items: center; height: 100vh; " }
+        .container { " max-width: 600px; background-color: #ffffff; border-radius: 4px; padding: 20px; box-shadow: 0px 4px 8px rgba(0, 0, 0, 0.1);  " }
+        .header { " text-align: center; margin-bottom: 20px; " }
+        .header h1 { " font-size: 36px; font-weight: 600; color: #0080ff; letter-spacing: 2px; margin: 20px 0;  " }
+        .logo { " display: block; margin: 0 auto; border: 0px; width: 125px; height: 120px; border-radius: 50%; box-shadow: 0px 4px 8px rgba(0, 0, 0, 0.2);         " }
+        .content { " color: #333333; font-family: 'Lato', Helvetica, Arial, sans-serif; font-size: 18px; font-weight: 400; line-height: 1.6; text-align: center; " }
+        .content strong { " font-weight: 600; color: #0080ff; " }
+        .footer { " padding-top: 30px; border-top: 1px solid #f5f5f5; text-align: center; " }
+        .footer p { "margin: 0; color: #666666; font-size: 16px;" }
+    </style>
+</head>
+<body>
+    <!-- HIDDEN PREHEADER TEXT -->
+    <div style="display: none; font-size: 1px; color: #fefefe; line-height: 1px; font-family: 'Lato', Helvetica, Arial, sans-serif; max-height: 0px; max-width: 0px; opacity: 0; overflow: hidden;">
+        Your Gateway to 3D Text Magic!
+    </div>
+    <div class="page-center">
+        <div class="container">
+            <!-- LOGO -->
+            <div class="header">
+                <h1>Welcome to vizAI</h1>
+                <img class="logo" src="https://i.ibb.co/G0t2czh/logo.jpg" alt="Logo" />
+            </div>
+            <div class="content">
+                <h3><strong>Your Gateway to 3D Text Magic!</strong></h3>
+                <p>Transform your ordinary text into extraordinary 3D visuals with vizAI! Our cutting-edge plugin opens up a whole new dimension,bringing your words to life and captivating your audience like never before. With mesmerizing 3D images that pop off the screen, ignite their imagination and create an immersive experience that leaves a lasting impression. Discover the power of vizAI and revolutionize the way you communicate with your audience.</p>
+                <p>Get ready to create captivating visual content with ease!</p>
+            </div>
+            <div class="footer">
+                <p>Thanks!<br>The vizAI Team</p>
+            </div>
+        </div>
+    </div>
+</body>
+</html>
+
+        '''
+    msg.attach(MIMEText(html, 'html'))
+
+    s.send_message(msg)
+    return "Success"
+
+def otp_gen():
+    digit="0123456789"
+    password=""
+    i=0
+    for i in range(6):
+        password=password+random.choice(digit)
+        i+1
+    file1 = open("myfile.txt","w")
+    file1.write(password)
+    file1.close()
+    return password
+#function used to generate 2d image from prompt
+def image_generator(description_1,counter):
+    model = joblib.load('filename.pkl')
+    autocast_model = joblib.load('torchpkl.pkl')
+    with autocast_model("cuda"):
+        image_1 = model(description_1).images[0]
+    image1 = image_1.save('./static/image'+str(counter)+'.jpg')
+
+#function used rendering login page
+@app.route('/',methods=["POST","GET"])
+def login():
+    if(request.method=="POST"):
+        username=request.form.get('email')
+        password=request.form.get('password')
+        query = "SELECT * FROM user_table WHERE username = %s AND password = %s"
+        values = (username,password)
+        mycursor.execute(query, values)
+        result = mycursor.fetchone()
+        if result:
+            return redirect(url_for('home'))
+        else:
+            return render_template("Login.html")
+        return render_template("Home.html")
+    if(request.method=="GET"):
+        return render_template("Login.html")
+#function used for rendering signup page
+@app.route('/signup',methods=["POST","GET"])
+def register():
+    if(request.method=="POST"):
+        username=request.form.get('email')
+        password=request.form.get('password')
+        opt=str(otp_gen())
+        msg=mail_send(opt,username)
+        query = "INSERT INTO user_table (username, password) VALUES (%s, %s)"
+        values = (username, password)
+        mycursor.execute(query, values)
+        mydb.commit()
+        return redirect('/')
+    if(request.method=="GET"):
+        return render_template("Register.html")
+#function used to render Text to 3d page
+@app.route('/text_to_3d',methods=["POST","GET"])
+def text_to_3d():
+    if(request.method=="GET"):
+        return render_template('TextTo3d.html')
+    if(request.method == "POST"):
+        prompt=request.form.get('prompt')
+        i=0
+        for i in range(0,4):
+            image_generator(prompt,i)
+        return redirect(url_for ('text_to_3d_preview'))
+#function used to render AR view of model
+@app.route('/ar_view',methods=["POST","GET"])
+def ar_view():
+    if(request.method=="GET"):
+        return render_template('AR.html')
+#function used to get list of images generated
+def get_image_filenames():
+    static_path = os.path.join(app.root_path, 'static/')
+    image_files = [f for f in os.listdir(static_path) if os.path.isfile(os.path.join(static_path, f))]
+    image_list = []
+    for filename in image_files:
+        full_path = os.path.join(static_path, filename)
+        image_list.append({'filename': filename, 'path': full_path})
+    return image_files
+#function used to preview the model generated
+@app.route('/text_to_3d_preview',methods=["POST","GET"])
+def text_to_3d_preview():
+    if(request.method=="GET"):
+        image_files = get_image_filenames()
+        return render_template('Preview2d.html',image_files=image_files)
+
+
+#function used to prompt for 3d model
+@app.route('/text_to_33d',methods=["POST","GET"])
+def text_to_33d():
+    if(request.method=="GET"):
+        return render_template('TextTo33d.html')
+    if(request.method == "POST"):
+        prompt=request.form.get('prompt')
+        model3d_generator(prompt)
+        return redirect(url_for ('text_to_33d_preview'))
+#function used to preview 3d model
+@app.route('/text_to_33d_preview',methods=["POST","GET"])
+def text_to_33d_preview():
+    if(request.method=="GET"):
+        return render_template('Preview3d.html')
+#function used to render about page
+@app.route('/home',methods=["POST","GET"])
+def home():
+    if(request.method=="GET"):
+        return render_template('Home.html')
+    if(request.method == "POST"):
+        return redirect(url_for('Home.html'))
+
+
+if __name__=='__main__':
+    app.run(port=5000,debug=True)

myfile.txt

@@ -0,0 +1 @@
+767834